pyabsa.framework.flag_class

Submodules

Package Contents

Classes

TaskNameOption

A dictionary subclass that maps task codes to task names.

TaskCodeOption

A class that defines task codes for various tasks.

LabelPaddingOption

A class that defines label padding options.

ModelSaveOption

A class that defines options for saving models.

ProxyAddressOption

A class that defines proxy address options.

DeviceTypeOption

A class that defines device type options.

Attributes

PyABSAMaterialHostAddress

class pyabsa.framework.flag_class.TaskNameOption[source]

Bases: dict

A dictionary subclass that maps task codes to task names.

code2name
get(key)[source]

Get the task name from the task code. :param key: The task code. :return: The task name.

class pyabsa.framework.flag_class.TaskCodeOption[source]

A class that defines task codes for various tasks.

Aspect_Polarity_Classification = 'APC'
Aspect_Term_Extraction_and_Classification = 'ATEPC'
Aspect_Sentiment_Triplet_Extraction = 'ASTE'
Sentiment_Analysis = 'TC'
Text_Classification = 'TC'
Text_Adversarial_Defense = 'TAD'
RNASequenceClassification = 'RNAC'
RNASequenceRegression = 'RNAR'
ProteinSequenceRegression = 'PR'
CodeDefectDetection = 'CDD'
Aspect_Category_Opinion_Sentiment_Triplet_Extraction = 'ACOS'
Universal_Sentiment_Analysis = 'USA'
class pyabsa.framework.flag_class.LabelPaddingOption[source]

A class that defines label padding options.

SENTIMENT_PADDING
LABEL_PADDING
class pyabsa.framework.flag_class.ModelSaveOption[source]

A class that defines options for saving models.

DO_NOT_SAVE_MODEL = 0
SAVE_MODEL_STATE_DICT = 1
SAVE_FULL_MODEL = 2
SAVE_FINE_TUNED_PLM = 3
class pyabsa.framework.flag_class.ProxyAddressOption[source]

A class that defines proxy address options.

CN_GITHUB_MIRROR = 'https://gitee.com/'
class pyabsa.framework.flag_class.DeviceTypeOption[source]

A class that defines device type options.

AUTO = True
CPU = 'cpu'
CUDA = 'cuda'
ALL_CUDA = 'allcuda'
pyabsa.framework.flag_class.PyABSAMaterialHostAddress = 'https://huggingface.co/spaces/yangheng/PyABSA/'[source]