From e57ffdbf7d9bbfa0a3b71ac010e429789c17ea17 Mon Sep 17 00:00:00 2001 From: Sudipta Basak Date: Tue, 18 Jul 2023 07:53:08 +1000 Subject: [PATCH] also need to remove from the quantilegb model --- uncoverml/optimise/models.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/uncoverml/optimise/models.py b/uncoverml/optimise/models.py index a122bba8..95579732 100644 --- a/uncoverml/optimise/models.py +++ b/uncoverml/optimise/models.py @@ -720,7 +720,7 @@ def __init__(self, target_transform='identity', loss='quantile', subsample=1.0, criterion='friedman_mse', min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0., max_depth=3, min_impurity_decrease=0., - min_impurity_split=None, init=None, random_state=None, + init=None, random_state=None, max_features=None, verbose=0, max_leaf_nodes=None, warm_start=False, validation_fraction=0.1, @@ -747,7 +747,6 @@ def __init__(self, target_transform='identity', loss='quantile', self.subsample = subsample self.max_features = max_features self.min_impurity_decrease = min_impurity_decrease - self.min_impurity_split = min_impurity_split self.random_state = random_state self.alpha = alpha self.verbose = verbose @@ -766,7 +765,6 @@ def __init__(self, target_transform='identity', loss='quantile', max_depth=max_depth, subsample=subsample, max_features=max_features, min_impurity_decrease=min_impurity_decrease, - min_impurity_split=min_impurity_split, random_state=random_state, verbose=verbose, max_leaf_nodes=max_leaf_nodes, warm_start=warm_start, validation_fraction=validation_fraction, @@ -781,7 +779,6 @@ def __init__(self, target_transform='identity', loss='quantile', max_depth=max_depth, subsample=subsample, max_features=max_features, min_impurity_decrease=min_impurity_decrease, - min_impurity_split=min_impurity_split, random_state=random_state, verbose=verbose, max_leaf_nodes=max_leaf_nodes, warm_start=warm_start, validation_fraction=validation_fraction, @@ -796,7 +793,6 @@ def __init__(self, target_transform='identity', loss='quantile', max_depth=max_depth, subsample=subsample, max_features=max_features, min_impurity_decrease=min_impurity_decrease, - min_impurity_split=min_impurity_split, random_state=random_state, verbose=verbose, max_leaf_nodes=max_leaf_nodes, warm_start=warm_start, validation_fraction=validation_fraction,