generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

GBT train/test was added to tree_engine sample

Browse Source
This commit is contained in:
Alexey Polovinkin 2010-10-17 19:18:42 +00:00
parent 1b1eab8e67
commit 0f7c2e016d
1 changed files with 5 additions and 0 deletions

5
samples/c/tree_engine.cpp

@ -37,6 +37,7 @@ int main()
CvBoost boost; CvBoost boost;
CvRTrees rtrees; CvRTrees rtrees;
CvERTrees ertrees; CvERTrees ertrees;
CvGBTrees gbtrees;
CvMLData data; CvMLData data;
@ -71,6 +72,10 @@ int main()
printf("======ERTREES=====\n"); printf("======ERTREES=====\n");
ertrees.train( &data, CvRTParams( 10, 2, 0, false, 16, 0, true, 0, 100, 0, CV_TERMCRIT_ITER )); ertrees.train( &data, CvRTParams( 10, 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() ); print_result( ertrees.calc_error( &data, CV_TRAIN_ERROR), ertrees.calc_error( &data, CV_TEST_ERROR ), ertrees.get_var_importance() );
printf("======GBTREES=====\n");
gbtrees.train( &data, CvGBTreesParams(CvGBTrees::DEVIANCE_LOSS, 100, 0.05f, 0.6f, 10, true));
print_result( gbtrees.calc_error( &data, CV_TRAIN_ERROR), gbtrees.calc_error( &data, CV_TEST_ERROR ), 0 );
} }
else else
printf("File can not be read"); printf("File can not be read");