From 5a41ebc44352135215059141d056cd4b2747c360 Mon Sep 17 00:00:00 2001 From: Bogdan Burlacu Date: Thu, 4 Apr 2024 15:08:54 +0200 Subject: [PATCH] sklearn.py: fix bug in the shape of the returned prediction --- pyoperon/sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyoperon/sklearn.py b/pyoperon/sklearn.py index fba9971..66375e9 100644 --- a/pyoperon/sklearn.py +++ b/pyoperon/sklearn.py @@ -596,4 +596,4 @@ def predict(self, X): Returns an array of ones. """ check_is_fitted(self) - return self.evaluate_model(self.model_, X).reshape(-1, 1) + return self.evaluate_model(self.model_, X).reshape(-1,)