ilyankou
/

spatial-classifier

Text Classification

sentence-transformers

spatial-queries

text-embeddings-inference

Model card Files Files and versions

ilyankou commited on Mar 5

Commit

e0a646c

·

verified ·

1 Parent(s): bf7a5da

Update README.md

Files changed (1) hide show

README.md +5 -3

README.md CHANGED Viewed

@@ -51,10 +51,12 @@ from setfit import SetFitModel
 model = SetFitModel.from_pretrained("ilyankou/spatial-classifier")
 preds = model([
-  "weather in erlanger ky",
-  "what is symptom of bipolar disorder"
 ])
-# => [1, 0]
 ```
 ## Training

 model = SetFitModel.from_pretrained("ilyankou/spatial-classifier")
 preds = model([
+  "nearest hospital",
+  "far from the truth",
+  "close to my heart",
+  "flood risk in this area"
 ])
+# => [1, 0, 0, 1]
 ```
 ## Training