fix error in processing new user data

david-cortes · Jul 28, 2019 · 34069fa · 34069fa
1 parent c3e3e58
commit 34069fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/hpfrec/__init__.py b/hpfrec/__init__.py
@@ -676,7 +676,7 @@ def _process_data_single(self, counts_df):
 					raise ValueError("Can only make calculations for items that were in the training set.")
 
 		counts_df['ItemId'] = counts_df.ItemId.values.astype(cython_loops.obj_ind_type)
-		counts_df['Count'] = counts_df.ItemId.values.astype(ctypes.c_float)
+		counts_df['Count'] = counts_df.Count.values.astype(ctypes.c_float)
 		return counts_df
 
 	def partial_fit(self, counts_df, batch_type='users', step_size=None,

diff --git a/setup.py b/setup.py
@@ -33,7 +33,7 @@ def build_extensions(self):
 	 'scipy',
 	 'cython'
 ],
-	version = '0.2.2.16',
+	version = '0.2.2.17',
 	description = 'Hierarchical Poisson matrix factorization for recommender systems',
 	author = 'David Cortes',
 	author_email = 'david.cortes.rivera@gmail.com',