Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -98,7 +98,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 98 |
|
| 99 |
X_train = sentence_embeddings.copy()
|
| 100 |
|
| 101 |
-
y_train = test_dataset["label"].copy()
|
| 102 |
|
| 103 |
#binary
|
| 104 |
y_train_binary = y_train.copy()
|
|
|
|
| 98 |
|
| 99 |
X_train = sentence_embeddings.copy()
|
| 100 |
|
| 101 |
+
y_train = np.array(test_dataset["label"].copy())
|
| 102 |
|
| 103 |
#binary
|
| 104 |
y_train_binary = y_train.copy()
|