Skip to content

Commit bd717a7

Browse files
authored
[FIX] IGCI Score (#76)
1 parent 39ea2b1 commit bd717a7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cdt/causality/pairwise/IGCI.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,8 +139,8 @@ def predict_proba(self, dataset, ref_measure='gaussian',
139139
"""
140140
a, b = dataset
141141
estimators = {'entropy': lambda x, y: eval_entropy(x) - eval_entropy(y), 'integral': integral_approx_estimator}
142-
ref_measures = {'gaussian': lambda x: standard_scale.fit_transform(x.reshape((-1, 1))),
143-
'uniform': lambda x: min_max_scale.fit_transform(x.reshape((-1, 1))), 'None': lambda x: x}
142+
ref_measures = {'gaussian': lambda x: standard_scale.fit_transform(x.reshape((-1, 1))).ravel(),
143+
'uniform': lambda x: min_max_scale.fit_transform(x.reshape((-1, 1))).ravel(), 'None': lambda x: x}
144144

145145
ref_measure = ref_measures[ref_measure]
146146
_estimator = estimators[estimator]

0 commit comments

Comments
 (0)