Spaces:
Build error
Build error
Commit
·
3fbc9ee
1
Parent(s):
4939d74
require
Browse files- scripts/evaluate.py +0 -1
scripts/evaluate.py
CHANGED
|
@@ -4,7 +4,6 @@ from sklearn.metrics import accuracy_score, f1_score
|
|
| 4 |
|
| 5 |
# Load dataset
|
| 6 |
dataset = load_dataset("allocine")["test"]
|
| 7 |
-
dataset["test"] = dataset["test"].select(range(5)) # Test on 200 samples
|
| 8 |
|
| 9 |
# Load model and tokenizer
|
| 10 |
model_path = "./models"
|
|
|
|
| 4 |
|
| 5 |
# Load dataset
|
| 6 |
dataset = load_dataset("allocine")["test"]
|
|
|
|
| 7 |
|
| 8 |
# Load model and tokenizer
|
| 9 |
model_path = "./models"
|