pyabsa.tasks.TextClassification.trainer.tc_trainer¶

Classes¶

TCTrainer

Trainer entry point for Text Classification.

Module Contents¶

class pyabsa.tasks.TextClassification.trainer.tc_trainer.TCTrainer(config: pyabsa.tasks.TextClassification.configuration.tc_configuration.TCConfigManager = None, dataset=None, from_checkpoint: str = None, checkpoint_save_mode: int = ModelSaveOption.SAVE_MODEL_STATE_DICT, auto_device: bool | str = DeviceTypeOption.AUTO, path_to_save=None, load_aug=False)¶

Bases: pyabsa.framework.trainer_class.trainer_template.Trainer

Trainer entry point for Text Classification.

Orchestrates dataset loading and the training instructor for text classifiers. After initialization, it launches the standard training pipeline. Use load_trained_model() to obtain a TextClassifier for inference.

training_instructor¶

inference_model_class¶