Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
057b382
1
Parent(s):
7f947b9
Experimenting with few-shot classification.
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ test_dataset = load_dataset("dair-ai/emotion", "split", split="test")
|
|
| 63 |
print(test_dataset)
|
| 64 |
classes = test_dataset.features["label"].names
|
| 65 |
print(classes)
|
| 66 |
-
train_dataset = get_templated_dataset()
|
| 67 |
print(train_dataset)
|
| 68 |
print(train_dataset[0])
|
| 69 |
|
|
|
|
| 63 |
print(test_dataset)
|
| 64 |
classes = test_dataset.features["label"].names
|
| 65 |
print(classes)
|
| 66 |
+
train_dataset = get_templated_dataset(candidate_labels=classes)
|
| 67 |
print(train_dataset)
|
| 68 |
print(train_dataset[0])
|
| 69 |
|