pyabsa.tasks._Archive.RNAClassification

Subpackages

Package Contents

Classes

RNACTrainer

Trainer class for training PyABSA models

RNACConfigManager

Simple object for storing attributes.

BERTRNACModelList

Built-in mutable sequence.

GloVeRNACModelList

Built-in mutable sequence.

RNACDatasetList

A list of available RNA datasets.

RNAClassificationDatasetList

A list of available RNA datasets.

RNAClassifier

Predictor

class pyabsa.tasks._Archive.RNAClassification.RNACTrainer(config: pyabsa.tasks._Archive.RNAClassification.configuration.rnac_configuration.RNACConfigManager = 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)[source]

Bases: pyabsa.framework.trainer_class.trainer_template.Trainer

Trainer class for training PyABSA models

class pyabsa.tasks._Archive.RNAClassification.RNACConfigManager(args, **kwargs)[source]

Bases: pyabsa.framework.configuration_class.configuration_template.ConfigManager

Simple object for storing attributes.

Implements equality by attribute names and values, and provides a simple string representation.

static set_rnac_config(configType: str, newitem: dict)[source]
static set_rnac_config_template(newitem)[source]
static set_rnac_config_base(newitem)[source]
static set_rnac_config_english(newitem)[source]
static set_rnac_config_chinese(newitem)[source]
static set_rnac_config_multilingual(newitem)[source]
static set_rnac_config_glove(newitem)[source]
static get_rnac_config_template()[source]
static get_rnac_config_base()[source]
static get_rnac_config_english()[source]
static get_rnac_config_chinese()[source]
static get_rnac_config_multilingual()[source]
static get_rnac_config_glove()[source]
class pyabsa.tasks._Archive.RNAClassification.BERTRNACModelList[source]

Bases: list

Built-in mutable sequence.

If no argument is given, the constructor creates a new empty list. The argument must be an iterable if specified.

BERT_MLP
class pyabsa.tasks._Archive.RNAClassification.GloVeRNACModelList[source]

Bases: list

Built-in mutable sequence.

If no argument is given, the constructor creates a new empty list. The argument must be an iterable if specified.

CNN
LSTM
Transformer
MHSA
class pyabsa.tasks._Archive.RNAClassification.RNACDatasetList[source]

Bases: list

A list of available RNA datasets.

class pyabsa.tasks._Archive.RNAClassification.RNAClassificationDatasetList[source]

Bases: RNACDatasetList

A list of available RNA datasets.

class pyabsa.tasks._Archive.RNAClassification.RNAClassifier(checkpoint=None, cal_perplexity=False, **kwargs)[source]

Bases: pyabsa.framework.prediction_class.predictor_template.InferenceModel

task_code
_log_write_args()[source]
batch_predict(target_file=None, print_result=True, save_result=False, ignore_error=True, **kwargs)[source]

Runs inference on a batch of data.

Parameters:

  • target_file (-) – str or Path. Path to the target file.

  • print_result (-) – bool. Whether to print the result.

  • save_result (-) – bool. Whether to save the result.

  • ignore_error (-) – bool. Whether to ignore errors.

  • kwargs (-) – dict. Additional keyword arguments.

Returns:

list. List of predictions.

Return type:

  • result

predict(text: str = None, print_result=True, ignore_error=True, **kwargs)[source]

Runs inference on a single sample.

Parameters:

  • text (-) – str. The text to predict.

  • print_result (-) – bool. Whether to print the result.

  • ignore_error (-) – bool. Whether to ignore errors.

  • kwargs (-) – dict. Additional keyword arguments.

Returns:

list. List of predictions.

Return type:

  • result

_run_prediction(save_path=None, print_result=True)[source]

This method should be implemented in the subclass for running predictions using the trained model.

Parameters:

kwargs – additional keyword arguments

Returns:

predicted labels or other prediction outputs

clear_input_samples()[source]
class pyabsa.tasks._Archive.RNAClassification.Predictor(checkpoint=None, cal_perplexity=False, **kwargs)[source]

Bases: RNAClassifier