gilf

/

english-yelp-sentiment

like 1

Hi. Does anyone know what sentiment each label represents? There are five labels, "LABEL_0", "LABEL_1", "LABEL_2", "LABEL_3", "LABEL_4". But it's unclear what each of them means.

Please check this notebook: https://github.com/onepointconsulting/yelp_bert/blob/master/simple_training/predict_pipeline.ipynb

The model is 5 star based. "LABEL_0" should be one star, "LABEL_4" 5 stars

BTW, if you want, you could change the id2label and label2id config attributes to switch them to 1/2/3/4/5 stars. Let me know if a PR would be welcome! 💪

@julien-c if you can do a pull request that would be most welcome.

see #2