pyabsa.tasks.AspectTermExtraction.trainer.atepc_trainer¶

Classes¶

ATEPCTrainer

Trainer entry point for Aspect Term Extraction and Polarity Classification (ATEPC).

Module Contents¶

class pyabsa.tasks.AspectTermExtraction.trainer.atepc_trainer.ATEPCTrainer(config: pyabsa.tasks.AspectTermExtraction.configuration.atepc_configuration.ATEPCConfigManager = 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 Aspect Term Extraction and Polarity Classification (ATEPC).

Connects configuration, datasets, and the ATEPC training instructor. After initialization, it launches the training routine. Use load_trained_model() to obtain an AspectExtractor for inference.

training_instructor¶

inference_model_class¶