From 66a3f527c80d179811624cc00e59efb531d2aecf Mon Sep 17 00:00:00 2001 From: Piotr Date: Fri, 30 Dec 2022 14:35:28 +0100 Subject: [PATCH] fix tests, replace mse with squared_error for tree based algorithms from sklearn (#595) --- supervised/preprocessing/label_encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervised/preprocessing/label_encoder.py b/supervised/preprocessing/label_encoder.py index 4df0e6c6..99f76299 100644 --- a/supervised/preprocessing/label_encoder.py +++ b/supervised/preprocessing/label_encoder.py @@ -50,5 +50,5 @@ def to_json(self): def from_json(self, data_json): keys = np.array(list(data_json.keys())) if len(keys) == 2 and "False" in keys and "True" in keys: - keys = [False, True] + keys = np.array([False, True]) self.lbl.classes_ = keys