pyabsa.framework.trainer_class.trainer_template
Module Contents
Classes
Trainer class for training PyABSA models |
Functions
|
- class pyabsa.framework.trainer_class.trainer_template.Trainer(config: pyabsa.framework.configuration_class.configuration_template.ConfigManager = None, dataset: Union[pyabsa.utils.data_utils.dataset_item.DatasetItem, pathlib.Path, str, pyabsa.framework.dataset_class.dataset_dict_class.DatasetDict] = None, from_checkpoint: Union[pathlib.Path, str] = None, checkpoint_save_mode: Union[pyabsa.framework.flag_class.flag_template.ModelSaveOption, int] = ModelSaveOption.SAVE_MODEL_STATE_DICT, auto_device: Union[str, bool] = DeviceTypeOption.AUTO, path_to_save: Union[pathlib.Path, str] = None, load_aug=False)[source]
Trainer class for training PyABSA models
- _run()[source]
just return the trained model for inference (e.g., polarity classification, aspect-term extraction)