Skip to content

Commit f84bd60

Browse files
authored
Merge pull request #112 from Jonathan-Salisbury/main
ANM incompatible with scikit-learn >= 1.1.3
2 parents ebcb1ac + aac327e commit f84bd60

File tree

1 file changed

+1
-1
lines changed
  • causallearn/search/FCMBased/ANM

1 file changed

+1
-1
lines changed

causallearn/search/FCMBased/ANM/ANM.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def fit_gp(self, X, y):
4949

5050
# fit Gaussian process, including hyperparameter optimization
5151
gpr.fit(X, y)
52-
pred_y = gpr.predict(X)
52+
pred_y = gpr.predict(X).reshape(-1, 1)
5353
return pred_y
5454

5555
def cause_or_effect(self, data_x, data_y):

0 commit comments

Comments
 (0)