diff --git a/mf.py b/mf.py index bddd9ce..9d30d3f 100644 --- a/mf.py +++ b/mf.py @@ -94,4 +94,4 @@ def full_matrix(self): """ Computer the full matrix using the resultant biases, P and Q """ - return self.b + self.b_u[:,np.newaxis] + self.b_i[np.newaxis:,] + self.P.dot(self.Q.T) + return self.b + self.b_u[:,np.newaxis] + self.b_i[np.newaxis, :] + self.P.dot(self.Q.T)