Spaces:
Sleeping
Sleeping
Commit ·
42328a2
1
Parent(s): 5584596
fixed model training error
Browse files- utils/rec_models.py +1 -1
utils/rec_models.py
CHANGED
|
@@ -179,7 +179,7 @@ def evaluate_recommendations(predicted_scores, train_matrix, test_df, user_index
|
|
| 179 |
true_items.add(item_idx)
|
| 180 |
|
| 181 |
# Collect actual vs predicted quantities for RMSE/R²
|
| 182 |
-
actual_quantity = row["
|
| 183 |
predicted_quantity = predicted_scores[user_idx, item_idx]
|
| 184 |
y_true.append(actual_quantity)
|
| 185 |
y_pred.append(predicted_quantity)
|
|
|
|
| 179 |
true_items.add(item_idx)
|
| 180 |
|
| 181 |
# Collect actual vs predicted quantities for RMSE/R²
|
| 182 |
+
actual_quantity = row["score"]
|
| 183 |
predicted_quantity = predicted_scores[user_idx, item_idx]
|
| 184 |
y_true.append(actual_quantity)
|
| 185 |
y_pred.append(predicted_quantity)
|