Skip to content
Permalink
Browse files

fix(encoder): fix PCAEncoder mean from fp64 to fp32

  • Loading branch information...
raccoonliukai committed Sep 9, 2019
1 parent f2fc7e7 commit 8357754abe143c8073196c70f561911a462db395
Showing with 1 addition and 1 deletion.
  1. +1 −1 gnes/encoder/numeric/pca.py
@@ -48,7 +48,7 @@ def train(self, vecs: np.ndarray, *args, **kwargs) -> None:
self.pca.partial_fit(vecs)

self.pca_components = np.transpose(self.pca.components_)
self.mean = self.pca.mean_
self.mean = self.pca.mean_.astype('float32')

@train_required
@batching

0 comments on commit 8357754

Please sign in to comment.
You can’t perform that action at this time.