From 0fcd7c4dba92e4eca5df853302b86c2f61924aad Mon Sep 17 00:00:00 2001 From: ShreeshaM07 Date: Thu, 4 Jul 2024 13:43:16 +0530 Subject: [PATCH] bins when 1D array in test params --- skpro/regression/binned/_sklearn_bin_regressor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/skpro/regression/binned/_sklearn_bin_regressor.py b/skpro/regression/binned/_sklearn_bin_regressor.py index 78c3548f1..a4fcd8635 100644 --- a/skpro/regression/binned/_sklearn_bin_regressor.py +++ b/skpro/regression/binned/_sklearn_bin_regressor.py @@ -262,9 +262,11 @@ def get_test_params(cls, parameter_set="default"): `create_test_instance` uses the first (or only) dictionary in `params` """ from sklearn.naive_bayes import GaussianNB + from sklearn.semi_supervised import LabelSpreading from sklearn.tree import DecisionTreeClassifier param1 = {"clf": DecisionTreeClassifier(), "bins": 4} param2 = {"clf": GaussianNB()} + params3 = {"clf": LabelSpreading(), "bins": [20, 80, 160, 250, 300, 380, 420]} - return [param1, param2] + return [param1, param2, params3]