Esempio n. 1
0
    def test_rfr(self):
        from lale.lib.sklearn import RandomForestRegressor

        reg = RandomForestRegressor(bootstrap=True, criterion='friedman_mse', max_depth=4, 
                max_features=0.9832410473940374, max_leaf_nodes=None, min_impurity_decrease=0.0, 
                min_impurity_split=None, min_samples_leaf=3, min_samples_split=2, min_weight_fraction_leaf=0.0, 
                n_estimators=29, n_jobs=4, oob_score=False, random_state=33, verbose=0, warm_start=False)        
        reg.fit(self.X_train, self.y_train)
Esempio n. 2
0
 def test_with_defaults(self):
     trainable = RandomForestRegressor()
     trained = trainable.fit(self.train_X, self.train_y)
     _ = trained.predict(self.test_X)