Skip to content

Commit

Permalink
Merge branch 'Dropout' into Dropout-opt
Browse files Browse the repository at this point in the history
  • Loading branch information
krivit committed May 31, 2024
2 parents 70be4e8 + 139b99e commit 85b3952
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions MCRS_GAT.py
Original file line number Diff line number Diff line change
Expand Up @@ -423,12 +423,12 @@ def evaluate_RS_Model_multiple_runs(fused_embeddings, user_id_map, item_id_map,
mean_rmse_std = np.mean(rmse_values)

# Print the standard deviation
print("Standard deviation of MAE over {} runs:".format(num_runs), mae_std)
print("Standard deviation of RMSE over {} runs:".format(num_runs), rmse_std)
print(f"Standard deviation of MAE over {num_runs} runs: {mae_std}")
print(f"Standard deviation of RMSE over {num_runs} runs: {rmse_std}")

# Print the mean of standard deviations
print("Mean of MAE over {} runs:".format(num_runs), mean_mae_std)
print("Mean of RMSE over {} runs:".format(num_runs), mean_rmse_std)
print(f"Mean of MAE over {num_runs} runs: {mean_mae_std}")
print(f"Mean of RMSE over {num_runs} runs: {mean_rmse_std}")

# Return the standard deviations
return mae_std, rmse_std
Expand Down

0 comments on commit 85b3952

Please sign in to comment.