-
Notifications
You must be signed in to change notification settings - Fork 204
Description
Hi, I repeat the emotion classification experiment and get terrible results. I couldn't what is the issue.
-
The experiment is repeated using the command line "!python3 experiments/run_clf_multihead.py --text-key Tweet --train data/semeval/train.csv --val data/semeval/val.csv --process-fn process_tweet".
-
Then, I got a series of classifiers in transformer_multihead from the 1)step.
-
Then I used "!python3 run_classifier.py --load transformer_multihead/model_ep0.clf --text-key Tweet --data data/semeval/val.csv --model transformer --write-results results/semeval/val_result.csv" on the validation set.
-
The performance is evaulated with respect to balanced accuracy, f1 score and ROC using metrics module from sklearn package. The results are shown as follows.
anger anticipation disgust fear joy sadness surprise trust
balanced accuracy 0.500876 0.500000 0.537070 0.500000 0.500000 0.500000 0.499412 0.500593
f1_score 0.525000 0.245545 0.488992 0.240318 0.622084 0.460469 0.000000 0.092672
ROC 0.537700 0.450639 0.549253 0.474326 0.508107 0.481694 0.504079 0.500841
Is anything I can do to make it work?
Regards, Yipeng