Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
8e9d450
1
Parent(s):
5557c1a
Experimenting with few-shot classification.
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ ort_model = ORTModelForFeatureExtraction.from_pretrained('BAAI/bge-small-en-v1.5
|
|
| 60 |
few_shot_model = SetFitModel.from_pretrained("moshew/bge-small-en-v1.5_setfit-sst2-english", multi_target_strategy="multi-output")
|
| 61 |
|
| 62 |
# Train few_shot_model
|
| 63 |
-
candidate_labels = ["supported", "refuted"]
|
| 64 |
reference_dataset = load_dataset("emotion")
|
| 65 |
dummy_dataset = Dataset.from_dict({})
|
| 66 |
train_dataset = get_templated_dataset(dummy_dataset, candidate_labels=candidate_labels, sample_size=8, template="This hypothesis is {}")
|
|
|
|
| 60 |
few_shot_model = SetFitModel.from_pretrained("moshew/bge-small-en-v1.5_setfit-sst2-english", multi_target_strategy="multi-output")
|
| 61 |
|
| 62 |
# Train few_shot_model
|
| 63 |
+
candidate_labels = ["supported", "refuted", "neither"]
|
| 64 |
reference_dataset = load_dataset("emotion")
|
| 65 |
dummy_dataset = Dataset.from_dict({})
|
| 66 |
train_dataset = get_templated_dataset(dummy_dataset, candidate_labels=candidate_labels, sample_size=8, template="This hypothesis is {}")
|