Skip to content

Facebook Research Aspect Based Sentiment Analysis

Notifications You must be signed in to change notification settings

ccj211985/IndoLEGO-ABSA

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Indo LEGO-ABSA

Indo LEGO-ABSA is an aspect-based sentiment analysis build on LEGO-ABSA framework for Indonesian language.

Prepare

You can create a conda environment or use your own python system environment. We use Python 3.9.16 during the experiment.

conda create --name indo-lego-absa python=3.9.16
pip install -r requirements.txt

Configurations

There are 4 configuration file in json format, located in the configs folder.

  • na_config: Path(s) to the Non-ABSA dataset (csv format) used in training. Consist of two columns, "input" and "output".
  • td_config: Containing configuration for the training data.
  • vd_config: Containing configuration for the validation data.
  • train_args: Training arguments, refer to https://huggingface.co/docs/transformers/main_classes/trainer for the details.

Training

You can use train.py or simple_train.py for simplicity.

For the arguments needed, you can use the help command.

python train.py --help

If you use simple_train.py, wrap the arguments in args.json.

python simple_train.py args.json

Inference

You can use the inference.ipynb notebook for inferencing.

About

Facebook Research Aspect Based Sentiment Analysis

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 68.8%
  • Python 31.2%