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

gnes-ai · Sep 9, 2019 · 8357754 · 8357754
1 parent f2fc7e7
commit 8357754
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gnes/encoder/numeric/pca.py b/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