From 89e3fde6ecb6a9c5096473202fd9a21c8ffe6cc6 Mon Sep 17 00:00:00 2001 From: Jonathan Salisbury Date: Mon, 1 May 2023 20:54:46 +0200 Subject: [PATCH] fixed bug with gpr array shape Signed-off-by: Jonathan Salisbury --- cdt/causality/pairwise/ANM.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cdt/causality/pairwise/ANM.py b/cdt/causality/pairwise/ANM.py index 4b1903bd..5b8e768e 100644 --- a/cdt/causality/pairwise/ANM.py +++ b/cdt/causality/pairwise/ANM.py @@ -186,7 +186,7 @@ def anm_score(self, x, y): float: ANM fit score """ gp = GaussianProcessRegressor().fit(x, y) - y_predict = gp.predict(x) + y_predict = gp.predict(x).reshape(-1, 1) indepscore = normalized_hsic(y_predict - y, x) return indepscore