Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -125,7 +125,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 125 |
logging.info(f"y_pred_bin:{y_pred_bin.shape}")
|
| 126 |
logging.info(f"y_pred_multi shape:{y_pred_multi.shape}")
|
| 127 |
|
| 128 |
-
y_pred_bin[
|
| 129 |
|
| 130 |
|
| 131 |
|
|
|
|
| 125 |
logging.info(f"y_pred_bin:{y_pred_bin.shape}")
|
| 126 |
logging.info(f"y_pred_multi shape:{y_pred_multi.shape}")
|
| 127 |
|
| 128 |
+
y_pred_bin[y_train_binary==1] = y_pred_multi
|
| 129 |
|
| 130 |
|
| 131 |
|