search

NVIDIA / sentiment-discovery

Unsupervised Language Modeling at scale for robust sentiment classification
Other
1.06k stars 202 forks source link

run_classifier for emotion classification? #66

Open ehsan-soe opened 4 years ago

ehsan-soe commented 4 years ago

Hi,

Is it possible to provide the run_classifier for emotion classification using your pretrained model?

pandeconscious commented 4 years ago

Use a csv file such as data/semeval/train.csv. In the following example, Tweet is the header of the column containing the actual text/tweet for which to find emotions. Download the emotion models as some dir such as path-to-downloaded-models

python3 run_classifier.py --load <path-to-downloaded-models/transformer_semeval.clf> --text-key Tweet --data data/semeval/train.csv --model transformer --write-results results.csv