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