diff --git a/cmfrec/__init__.py b/cmfrec/__init__.py index 7f612b2..fdc7191 100644 --- a/cmfrec/__init__.py +++ b/cmfrec/__init__.py @@ -1972,10 +1972,10 @@ def _item_factors_cold(self, I=None, I_bin=None, I_col=None, I_val=None): if isinstance(self.l1_lambda, np.ndarray): l1_lambda = self.l1_lambda[3] - lambda_bias = self.l1_lambda[1] + l1_lambda_bias = self.l1_lambda[1] else: l1_lambda = self.l1_lambda - lambda_bias = self.l1_lambda + l1_lambda_bias = self.l1_lambda I, I_col, I_val, I_bin = self._process_new_U(U=I, U_col=I_col, U_val=I_val, U_bin=I_bin, is_I=True) @@ -1997,7 +1997,7 @@ def _item_factors_cold(self, I=None, I_bin=None, I_col=None, I_val=None): self.A_, np.empty((0,0), dtype=self.dtype_), self.D_, - self.D_bin_, + self.Dbin_, self.Ai_, np.empty((0,0), dtype=self.dtype_), np.empty((0,0), dtype=self.dtype_), diff --git a/setup.py b/setup.py index 9ecc254..26a1297 100644 --- a/setup.py +++ b/setup.py @@ -348,7 +348,7 @@ def test_supports_clang_reassociate(self): setup( name = "cmfrec", packages = ["cmfrec"], - version = '3.5.1-5', + version = '3.5.1-6', description = 'Collective matrix factorization', author = 'David Cortes', url = 'https://github.com/david-cortes/cmfrec',