diff --git a/pqkmeans/encoder/pq_encoder.py b/pqkmeans/encoder/pq_encoder.py index 513b9fb..9ac922a 100644 --- a/pqkmeans/encoder/pq_encoder.py +++ b/pqkmeans/encoder/pq_encoder.py @@ -22,9 +22,9 @@ def fit(self, x_train): self.Ds = int(D / self.M) assert self.trained_encoder is None, "fit must be called only once" - codewords = numpy.zeros((self.M, self.Ks, self.Ds), dtype=numpy.float) + codewords = numpy.zeros((self.M, self.Ks, self.Ds), dtype=float) for m in range(self.M): - x_train_sub = x_train[:, m * self.Ds: (m + 1) * self.Ds].astype(numpy.float) + x_train_sub = x_train[:, m * self.Ds: (m + 1) * self.Ds].astype(float) codewords[m], _ = kmeans2(x_train_sub, self.Ks, iter=self.iteration, minit='points') self.trained_encoder = TrainedPQEncoder(codewords, self.code_dtype) @@ -66,7 +66,7 @@ def decode_multi(self, codes): assert M == self.M assert codes.dtype == self.code_dtype - decoded = numpy.empty((N, self.Ds * self.M), dtype=numpy.float) + decoded = numpy.empty((N, self.Ds * self.M), dtype=float) for m in range(self.M): decoded[:, m * self.Ds: (m + 1) * self.Ds] = self.codewords[m][codes[:, m], :] return decoded diff --git a/requirements.txt b/requirements.txt index 53751e4..8c90c38 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -numpy<1.21 +numpy scikit-learn pipe<2.0 scipy