/** * Called after an iteration has finished. */ void after_iteration(int iteration, CE_Graph_context &gcontext) { svdpp.itmFctrStep *= svdpp.step_dec; svdpp.itmFctr2Step *= svdpp.step_dec; svdpp.usrFctrStep *= svdpp.step_dec; svdpp.itmBiasStep *= svdpp.step_dec; svdpp.usrBiasStep *= svdpp.step_dec; training_rmse(iteration, gcontext); validation_rmse(&svdpp_predict, gcontext); }
/** * Called after an iteration has finished. */ void after_iteration(int iteration, graphchi_context &gcontext) { if (iteration == pmf_burn_in){ printf("Finished burn-in period. starting to aggregate samples\n"); } if (pmf_additional_output && iiter >= pmf_burn_in){ char buf[256]; sprintf(buf, "%s-%d", training.c_str(), iiter-pmf_burn_in); output_pmf_result(buf); } double res = training_rmse(iteration, gcontext); sample_hyperpriors(res); rmse_index = 0; rmse_type = VALIDATION; validation_rmse(&pmf_predict, gcontext, 3, &validation_avgprod, pmf_burn_in); if (iteration >= pmf_burn_in){ rmse_index = 0; rmse_type = TEST; test_predictions(&pmf_predict, &gcontext, iiter == niters-1, &test_avgprod); } iiter++; }
void after_iteration(int iteration, graphchi_context &gcontext) { training_rmse(iteration, gcontext, algo == ITEM_MEAN); validation_rmse(&baseline_predict, gcontext); }