|
|
|
@ -121,7 +121,7 @@ int main(int argc, char** argv) |
|
|
|
|
print_result( rtrees.calc_error( &data, CV_TRAIN_ERROR), rtrees.calc_error( &data, CV_TEST_ERROR ), rtrees.get_var_importance() ); |
|
|
|
|
|
|
|
|
|
printf("======ERTREES=====\n"); |
|
|
|
|
ertrees.train( &data, CvRTParams( 10, 2, 0, false, 16, 0, true, 0, 100, 0, CV_TERMCRIT_ITER )); |
|
|
|
|
ertrees.train( &data, CvRTParams( 18, 2, 0, false, 16, 0, true, 0, 100, 0, CV_TERMCRIT_ITER )); |
|
|
|
|
print_result( ertrees.calc_error( &data, CV_TRAIN_ERROR), ertrees.calc_error( &data, CV_TEST_ERROR ), ertrees.get_var_importance() ); |
|
|
|
|
|
|
|
|
|
printf("======GBTREES=====\n"); |
|
|
|
|