MarieAngeA13 commited on
Commit
27c5f6a
·
1 Parent(s): 70adb73

Update Sentiment_analysis_with_bert.py

Browse files
Files changed (1) hide show
  1. Sentiment_analysis_with_bert.py +7 -1
Sentiment_analysis_with_bert.py CHANGED
@@ -518,4 +518,10 @@ def generate_improved_text(review_text, predicted_sentiment):
518
  else:
519
  improved_text = review_text
520
 
521
- return improved_text
 
 
 
 
 
 
 
518
  else:
519
  improved_text = review_text
520
 
521
+ return improved_text
522
+
523
+ from transformers import BertModel
524
+
525
+ bert_model = BertModel.from_pretrained('bert-base-uncased')
526
+
527
+ bert_model.save_pretrained('C:/Users/Marie-Ange/Downloads')