Spaces:
Running
Running
Update tasks/text.py
Browse files- tasks/text.py +3 -0
tasks/text.py
CHANGED
|
@@ -122,6 +122,9 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 122 |
|
| 123 |
y_pred_multi = xgb_multi.predict(X_train_multi.reshape(-1,768)) + 1
|
| 124 |
|
|
|
|
|
|
|
|
|
|
| 125 |
y_pred_bin[y_pred_bin==1] = y_pred_multi
|
| 126 |
|
| 127 |
|
|
|
|
| 122 |
|
| 123 |
y_pred_multi = xgb_multi.predict(X_train_multi.reshape(-1,768)) + 1
|
| 124 |
|
| 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_pred_bin==1] = y_pred_multi
|
| 129 |
|
| 130 |
|