Spaces:

joonkim
/

bert-political-sentiment-analysis

Running

joonkim commited on Jan 23, 2023

Commit

4fe9941

•

1 Parent(s): b1ca1db

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -7,7 +7,7 @@ class SentimentClassifier(nn.Module) :
     super(SentimentClassifier, self).__init__()
-    self.BERT = BertModel.from_pretrained("bert-base-cased")
     self.dropout = nn.Dropout(p=0.3)
     self.fc = nn.Linear(self.BERT.config.hidden_size, n_classes)

     super(SentimentClassifier, self).__init__()
+    self.BERT = BertModel.from_pretrained('bert-base-uncased')
     self.dropout = nn.Dropout(p=0.3)
     self.fc = nn.Linear(self.BERT.config.hidden_size, n_classes)