Exemple #1
0
  /**
   * 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);
  }
Exemple #2
0
  /**
   * 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);
 }