ragarwal's picture
Initial README
1e6518e
|
raw
history blame
503 Bytes
metadata
license: mit

Trained Using Sentence-Transformers

Usage

sentence = "During its monthly call, the National Oceanic and Atmospheric Administration warned of increased temperatures and low precipitation" 
labels = ["Computer", "Climate Change", "Tablet", "Football", "Artificial Intelligence", "Global Warming"] 

scores = model.predict([[sentence, l] for l in labels])
print(scores) 
#array([0.04118565, 0.2435827 , 0.03941465, 0.00203637, 0.00501176, 0.1423797], dtype=float32)