void defaultGBMs ( ) { Algo algo = Algo . GBM ; WorkAllocations . Work work = workAllocations . getAllocation ( algo , JobType . ModelBuild ) ; if ( work == null ) return ; Job gbmJob ; GBMParameters gbmParameters = new GBMParameters ( ) ; setCommonModelBuilderParams ( gbmParameters ) ; gbmParameters . _score_tree_interval = 5 ; gbmParameters . _histogram_type = SharedTreeParameters . HistogramType . AUTO ; gbmParameters . _ntrees = 1000 ; gbmParameters . _sample_rate = 0.8 ; gbmParameters . _col_sample_rate = 0.8 ; gbmParameters . _col_sample_rate_per_tree = 0.8 ; // Default 1: gbmParameters . _max_depth = 6 ; gbmParameters . _min_rows = 1 ; gbmJob = trainModel ( null , work , gbmParameters ) ; pollAndUpdateProgress ( Stage . ModelTraining , "GBM 1" , work , this . job ( ) , gbmJob ) ; // Default 2: gbmParameters . _max_depth = 7 ; gbmParameters . _min_rows = 10 ; gbmJob = trainModel ( null , work , gbmParameters ) ; pollAndUpdateProgress ( Stage . ModelTraining , "GBM 2" , work , this . job ( ) , gbmJob ) ; // Default 3: gbmParameters . _max_depth = 8 ; gbmParameters . _min_rows = 10 ; gbmJob = trainModel ( null , work , gbmParameters ) ; pollAndUpdateProgress ( Stage . ModelTraining , "GBM 3" , work , this . job ( ) , gbmJob ) ; // Default 4: gbmParameters . _max_depth = 10 ; gbmParameters . _min_rows = 10 ; gbmJob = trainModel ( null , work , gbmParameters ) ; pollAndUpdateProgress ( Stage . ModelTraining , "GBM 4" , work , this . job ( ) , gbmJob ) ; // Default 5: gbmParameters . _max_depth = 15 ; gbmParameters . _min_rows = 100 ; gbmJob = trainModel ( null , work , gbmParameters ) ; pollAndUpdateProgress ( Stage . ModelTraining , "GBM 5" , work , this . job ( ) , gbmJob ) ; }
Build Arno s magical 5 default GBMs .