aliarafat-stack-ml commited on
Commit
42328a2
·
1 Parent(s): 5584596

fixed model training error

Browse files
Files changed (1) hide show
  1. 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["Quantity"]
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)