VityaVitalich
/

bert-tiny-sst2

Text Classification

Generated from Trainer

Model card Files Files and versions Community

VityaVitalich commited on Oct 2, 2023

Commit

2e14b76

·

1 Parent(s): 1536c4b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ from transformers import BertTokenizer, BertForSequenceClassification, TrainingA
 import datasets
 model = BertForSequenceClassification.from_pretrained('VityaVitalich/bert-tiny-sst2')
-tokenizer = BertTokenizer.from_pretrained('M-FAC/bert-tiny-finetuned-sst2')
 def create_data(tokenizer):
   train_set = datasets.load_dataset('sst2', split='train').remove_columns(['idx'])

 import datasets
 model = BertForSequenceClassification.from_pretrained('VityaVitalich/bert-tiny-sst2')
+tokenizer = BertTokenizer.from_pretrained('VityaVitalich/bert-tiny-sst2')
 def create_data(tokenizer):
   train_set = datasets.load_dataset('sst2', split='train').remove_columns(['idx'])