tlt.models.text_classification.text_classification_model.TextClassificationModel.train

abstract TextClassificationModel.train(dataset: BaseDataset, output_dir, epochs=1, initial_checkpoints=None, do_eval=True)

Train the model using the specified dataset