From cd75930ac8285967a82852424c359c0f25edc7af Mon Sep 17 00:00:00 2001 From: Roman Zavodskikh Date: Sun, 2 Aug 2020 20:22:32 +0300 Subject: [PATCH] Pass explicitly specified scoring --- hunga_bunga/classification.py | 2 +- hunga_bunga/regression.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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