From 6280ed383d3ddeb563c3455abc5e87baab5bd185 Mon Sep 17 00:00:00 2001 From: Sara Iris Garcia Date: Fri, 9 Aug 2019 14:40:02 -0600 Subject: [PATCH] Update iteritems() to python3 compatibility --- pykliep.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pykliep.py b/pykliep.py index a0859f7..d94395c 100644 --- a/pykliep.py +++ b/pykliep.py @@ -1,6 +1,6 @@ import numpy as np import warnings - +from future.utils import iteritems class DensityRatioEstimator: """ Class to accomplish direct density estimation implementing the original KLIEP @@ -83,7 +83,7 @@ def fit(self, X_train, X_test, alpha_0=None): sigma=sigma) j_scores[(num_param,sigma)] = np.mean(j_scores[(num_param,sigma)]) - sorted_scores = sorted([x for x in j_scores.iteritems() if np.isfinite(x[1])], key=lambda x :x[1], reverse=True) + sorted_scores = sorted([x for x in iteritems(j_scores) if np.isfinite(x[1])], key=lambda x :x[1], reverse=True) if len(sorted_scores) == 0: warnings.warn('LCV failed to converge for all values of sigma.') return self