Merge pull request #512 from PaschalisSk/hotfix/predict-returned-type

LightFM.predict returning np.float64 instead of np.float32
lyst · Nov 27, 2020 · 445a91c · 445a91c
2 parents 74512fb + e1ed5a7
commit 445a91c
Show file tree

Hide file tree

Showing 4 changed files with 8,140 additions and 7,712 deletions.
diff --git a/lightfm/_lightfm_fast.pyx.template b/lightfm/_lightfm_fast.pyx.template
@@ -1186,7 +1186,7 @@ def predict_lightfm(CSRMatrix item_features,
                     CSRMatrix user_features,
                     int[::1] user_ids,
                     int[::1] item_ids,
-                    double[::1] predictions,
+                    flt[::1] predictions,
                     FastLightFM lightfm,
                     int num_threads):
     """