diff --git a/hunga_bunga/classification.py b/hunga_bunga/classification.py index 7871f36..a1573c1 100644 --- a/hunga_bunga/classification.py +++ b/hunga_bunga/classification.py @@ -217,7 +217,7 @@ def __init__(self, brain=False, test_size = 0.2, n_splits = 5, random_state=None self.n_splits = n_splits self.random_state = random_state self.upsample = upsample - self.scoring = None + self.scoring = scoring self.verbose = verbose self.n_jobs = n_jobs self.normalize_x = normalize_x diff --git a/hunga_bunga/regression.py b/hunga_bunga/regression.py index 0a9967a..87f6409 100644 --- a/hunga_bunga/regression.py +++ b/hunga_bunga/regression.py @@ -333,7 +333,7 @@ def __init__(self, brain=False, test_size = 0.2, n_splits = 5, random_state=None self.n_splits = n_splits self.random_state = random_state self.upsample = upsample - self.scoring = None + self.scoring = scoring self.verbose = verbose self.n_jobs = n_jobs self.normalize_x = normalize_x