transformers

Transformers tools.

class transformer_tools.transformers.KFoldLabeledDataset(n_splits=7, n_repeats=1, random_state=None)[source]

Bases: object

Utility to do k-fold cross-validation on LabeledDataset.

split(labeled_dataset, stratification_labels=None)[source]

Generates data splits of training and test set.

class transformer_tools.transformers.LabeledDataset(encodings, labels)[source]

Bases: Dataset

Dataset with labes.