model.predict was returning np.float64 instead of np.float32

lyst · Feb 6, 2020 · 6f47aae · 6f47aae
1 parent 1f3f35d
commit 6f47aae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightfm/lightfm.py b/lightfm/lightfm.py
@@ -813,7 +813,7 @@ def predict(
 
         lightfm_data = self._get_lightfm_data()
 
-        predictions = np.empty(len(user_ids), dtype=np.float64)
+        predictions = np.empty(len(user_ids), dtype=np.float32)
 
         predict_lightfm(
             CSRMatrix(item_features),