File size: 483 Bytes
6b56e9f |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
model_name = "textattack/bert-base-uncased-rotten-tomatoes"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
text = "This is a positive review."
inputs = tokenizer(text, return_tensors="tf")
outputs = model(inputs)
scores = tf.nn.softmax(outputs.logits, axis=1).numpy()[0]
positive_score = scores[1]
negative_score = scores[0]
|