Skip to content

Commit 03ec379

Browse files
authored
Merge pull request #30 from rolisz/patch-2
Remove itemfreq
2 parents 4ed37d7 + c9b0b24 commit 03ec379

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

eli5/lime/samplers.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import six
77

88
import numpy as np
9-
from scipy.stats import itemfreq
109
from sklearn.base import BaseEstimator, clone
1110
from sklearn.neighbors import KernelDensity
1211
from sklearn.metrics import pairwise_distances
@@ -188,7 +187,7 @@ def _sampler_n_samples(self, n_samples):
188187
p=self.weights)
189188
return [
190189
(self.samplers[idx], freq)
191-
for idx, freq in itemfreq(sampler_indices)
190+
for idx, freq in zip(*np.unique(sampler_indices, return_counts=True))
192191
]
193192

194193

0 commit comments

Comments
 (0)