diff --git a/build/lib/pymatch/Matcher.py b/build/lib/pymatch/Matcher.py index bf4537f..45adbcc 100644 --- a/build/lib/pymatch/Matcher.py +++ b/build/lib/pymatch/Matcher.py @@ -523,4 +523,5 @@ def assign_weight_vector(self): @staticmethod def _scores_to_accuracy(m, X, y): preds = [[1.0 if i >= .5 else 0.0 for i in m.predict(X)]] - return (y == preds).sum() * 1.0 / len(y) + # return (y == preds).sum() * 1.0 / len(y) + return (y.to_numpy().T == preds).sum() * 1.0 / len(y)