Commit ·
26d7d1f
1
Parent(s): 14c7afe
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
classifier_2 = BERTForClassification(bert_model=model, num_classes=6)
|
| 2 |
+
|
| 3 |
+
classifier_2.load_weights('/content/drive/MyDrive/FineTunning2/model2.1/model')
|
| 4 |
+
|
| 5 |
+
example_input = "Your work is very very bad"
|
| 6 |
+
|
| 7 |
+
encoded_example = tokenizer.encode(example_input, padding=True, truncation=True, return_tensors="tf")
|
| 8 |
+
|
| 9 |
+
p = classifier_2(encoded_example)
|