1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
--- a/umap/umap_.py
+++ b/umap/umap_.py
@@ -2375,9 +2375,9 @@
raise ValueError("Non-zero distances from samples to themselves!")
if self.knn_dists is None:
self._knn_indices = np.zeros(
- (X.shape[0], self.n_neighbors), dtype=np.int
+ (X.shape[0], self.n_neighbors), dtype=int
)
- self._knn_dists = np.zeros(self._knn_indices.shape, dtype=np.float)
+ self._knn_dists = np.zeros(self._knn_indices.shape, dtype=float)
for row_id in range(X.shape[0]):
# Find KNNs row-by-row
row_data = X[row_id].data
|
