Skip to content

Commit

Permalink
pass unittest
Browse files Browse the repository at this point in the history
  • Loading branch information
jason.yoon authored and lucas(김광섭) committed Aug 29, 2019
1 parent 98844f4 commit d64fad7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion buffalo/algo/als.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,12 +74,12 @@ def normalize(self, group='item'):
self.opt._nrz_P = True

def initialize(self):
self.vdim = self.obj.get_vdim() if self.opt.accelerator else self.opt.d
super().initialize()
self.init_factors()

def init_factors(self):
assert self.data, 'Data is not setted'
self.vdim = self.obj.get_vdim() if self.opt.accelerator else self.opt.d
header = self.data.get_header()
for name, rows in [('P', header['num_users']), ('Q', header['num_items'])]:
setattr(self, name, None)
Expand Down

0 comments on commit d64fad7

Please sign in to comment.