From 0f7c2e016deb2e7b78152d2c60e9dbecb2d134e0 Mon Sep 17 00:00:00 2001 From: Alexey Polovinkin Date: Sun, 17 Oct 2010 19:18:42 +0000 Subject: [PATCH] GBT train/test was added to tree_engine sample --- samples/c/tree_engine.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/samples/c/tree_engine.cpp b/samples/c/tree_engine.cpp index add1bbd..0bd6bf4 100644 --- a/samples/c/tree_engine.cpp +++ b/samples/c/tree_engine.cpp @@ -37,6 +37,7 @@ int main() CvBoost boost; CvRTrees rtrees; CvERTrees ertrees; + CvGBTrees gbtrees; CvMLData data; @@ -71,6 +72,10 @@ int main() printf("======ERTREES=====\n"); 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() ); + + 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 printf("File can not be read"); -- 2.7.4