File size: 334 Bytes
26d7d1f |
1 2 3 4 5 6 7 8 9 |
classifier_2 = BERTForClassification(bert_model=model, num_classes=6)
classifier_2.load_weights('/content/drive/MyDrive/FineTunning2/model2.1/model')
example_input = "Your work is very very bad"
encoded_example = tokenizer.encode(example_input, padding=True, truncation=True, return_tensors="tf")
p = classifier_2(encoded_example) |