pyabsa.tasks.ABSAInstruction

Submodules

Package Contents

Classes

ABSAGenerator

ACOSDatasetList

The following datasets are for aspect polarity classification task.

InstructDatasetLoader

class pyabsa.tasks.ABSAInstruction.ABSAGenerator(checkpoint)[source]

Bases: T5Generator

class pyabsa.tasks.ABSAInstruction.ACOSDatasetList[source]

Bases: list

The following datasets are for aspect polarity classification task. The datasets are collected from different sources, you can use the id to locate the dataset.

Laptop14
Restaurant14
Restaurant15
Restaurant16
Chinese_Zhang
Synthetic
class pyabsa.tasks.ABSAInstruction.InstructDatasetLoader(train_df_id, test_df_id, train_df_ood=None, test_df_ood=None, sample_size=1)[source]
prepare_instruction_dataloader(df)[source]

Prepare the data in the input format required.

create_datasets(tokenize_function)[source]