minor changes
This commit is contained in:
parent
23a9b7bb9c
commit
098795cd16
@ -24,7 +24,7 @@ vector<Scalar> classColors;
|
||||
#define SVM 0 // support vectors machine
|
||||
#define DT 1 // decision tree
|
||||
#define BT 0 // ADA Boost
|
||||
#define GBT 1 // gradient boosted trees
|
||||
#define GBT 0 // gradient boosted trees
|
||||
#define RF 0 // random forest
|
||||
#define ERT 0 // extremely randomized trees
|
||||
#define ANN 0 // artificial neural networks
|
||||
@ -284,7 +284,7 @@ void find_decision_boundary_GBT()
|
||||
0.05f, // shrinkage
|
||||
0.6f, // subsample_portion
|
||||
2, // max_depth
|
||||
true // use_surrogates )
|
||||
false // use_surrogates )
|
||||
);
|
||||
|
||||
gbtrees.train( trainSamples, CV_ROW_SAMPLE, trainClasses, Mat(), Mat(), var_types, Mat(), params );
|
||||
@ -555,31 +555,31 @@ int main()
|
||||
|
||||
#if DT
|
||||
find_decision_boundary_DT();
|
||||
namedWindow( "DT", 1 );
|
||||
namedWindow( "DT", WINDOW_AUTOSIZE );
|
||||
imshow( "DT", imgDst );
|
||||
#endif
|
||||
|
||||
#if BT
|
||||
find_decision_boundary_BT();
|
||||
namedWindow( "BT", 1 );
|
||||
namedWindow( "BT", WINDOW_AUTOSIZE );
|
||||
imshow( "BT", imgDst);
|
||||
#endif
|
||||
|
||||
#if GBT
|
||||
find_decision_boundary_GBT();
|
||||
namedWindow( "GBT", 1 );
|
||||
namedWindow( "GBT", WINDOW_AUTOSIZE );
|
||||
imshow( "GBT", imgDst);
|
||||
#endif
|
||||
|
||||
#if RF
|
||||
find_decision_boundary_RF();
|
||||
namedWindow( "RF", 1 );
|
||||
namedWindow( "RF", WINDOW_AUTOSIZE );
|
||||
imshow( "RF", imgDst);
|
||||
#endif
|
||||
|
||||
#if ERT
|
||||
find_decision_boundary_ERT();
|
||||
namedWindow( "ERT", 1 );
|
||||
namedWindow( "ERT", WINDOW_AUTOSIZE );
|
||||
imshow( "ERT", imgDst);
|
||||
#endif
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user