diff options
| -rw-r--r-- | Wasserstein_Distance.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Wasserstein_Distance.py b/Wasserstein_Distance.py index 5965ed5..08439d2 100644 --- a/Wasserstein_Distance.py +++ b/Wasserstein_Distance.py | |||
| @@ -138,7 +138,7 @@ class Wasserstein_Retriever(KNeighborsClassifier): | |||
| 138 | dist = self._pairwise_wmd(X) | 138 | dist = self._pairwise_wmd(X) |
| 139 | return super(Wasserstein_Retriever, self).kneighbors(dist, n_neighbors) | 139 | return super(Wasserstein_Retriever, self).kneighbors(dist, n_neighbors) |
| 140 | 140 | ||
| 141 | def align(self, X, n_neighbors=1) | 141 | def align(self, X, n_neighbors=1): |
| 142 | """ | 142 | """ |
| 143 | Wrapper function over kneighbors to return | 143 | Wrapper function over kneighbors to return |
| 144 | precision at one and percentage values | 144 | precision at one and percentage values |