Allow pickle when loading model

redbubble · Nov 18, 2019 · b5c80d6 · b5c80d6
1 parent efccdf0
commit b5c80d6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/pyrec/implicit/als.py b/pyrec/implicit/als.py
@@ -116,7 +116,7 @@ def save(self, base_name, user_factors=False, compress=False):
 
 def load_recommender(als_model_file: str, index_file: str, item_feature_file: str = None, **kwargs) -> ImplicitRecommender:
     log.info("Loading als model")
-    data = np.load(als_model_file)
+    data = np.load(als_model_file, allow_pickle=True)
     model = AlternatingLeastSquares(factors=data['model.item_factors'].shape[1])
     model.item_factors = data['model.item_factors']
     model.YtY  # This will initialize the _YtY instance variable which is used directly in internal methods