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

brief gbt documentation added. some sample fixes made. code updated.

Browse Source
This commit is contained in:
P. Druzhkov
2011-06-15 21:54:25 +00:00
parent 9c071c6a30
commit e20d570ed1
5 changed files with 668 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
samples/c/tree_engine.cpp
View File

@@ -125,7 +125,10 @@ int main(int argc, char** argv)
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));
if (categorical_response)
gbtrees.train( &data, CvGBTreesParams(CvGBTrees::DEVIANCE_LOSS, 100, 0.1f, 0.8f, 5, false));
else
gbtrees.train( &data, CvGBTreesParams(CvGBTrees::SQUARED_LOSS, 100, 0.1f, 0.8f, 5, false));
print_result( gbtrees.calc_error( &data, CV_TRAIN_ERROR), gbtrees.calc_error( &data, CV_TEST_ERROR ), 0 ); //doesn't compute importance
}
else