ndiy commited on
Commit
a874f18
1 Parent(s): d08bb4e

add config

Browse files
Files changed (1) hide show
  1. config.py +8 -0
config.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+
2
+ base_model = 'mrcaelumn/yelp_restaurant_review_sentiment_analysis'
3
+ senti_map = {'LABEL_2': 'Positive', 'LABEL_0': 'Negative', 'LABEL_1': 'Neutral'}
4
+
5
+ absa_model = ("tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-aspect",
6
+ "tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-polarity")
7
+
8
+ senti_color = {'positive': 'aqua', 'neutral': 'gray', 'negative': 'orange'}