Update TextAnalysisSpacy.py

parent e289393a
Showing with 2 additions and 2 deletions
...@@ -308,8 +308,8 @@ class TextAnalysisSpacy(): ...@@ -308,8 +308,8 @@ class TextAnalysisSpacy():
return self.df return self.df
def emotions(self): def emotions(self):
classifier6 = pipeline("text-classification",model='model1_path', return_all_scores=True) classifier6 = pipeline("text-classification",model='bhadresh-savani/distilbert-base-uncased-emotion', return_all_scores=True)
classifier27 = pipeline("text-classification",model='model2_path', return_all_scores=True) classifier27 = pipeline("text-classification",model='bhadresh-savani/bert-base-go-emotion', return_all_scores=True)
i=0 i=0
for text in tqdm(self.df['text']): for text in tqdm(self.df['text']):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment