01032a5db7f237fe23cce1a7700c748b38ff1077,algorithm/rating/EE.py,EE,buildModel,#EE#,22

Before Change

            for entry in self.dao.trainingData:
                user, item, rating = entry
                error = rating - self.predict(user,item)
                u = self.dao.getUserId(user)
                i = self.dao.getItemId(item)
                self.loss += error ** 2
                self.loss += self.regU * (self.X[u] - self.Y[i]).dot(self.X[u] - self.Y[i])
                bu = self.Bu[u]
                bi = self.Bi[i]
                self.loss += self.regB * bu ** 2 + self.regB * bi ** 2
                // update latent vectors
                self.X[u] -= self.lRate * (error + self.regU) * (self.X[u] - self.Y[i])
                self.Y[i] += self.lRate * (error + self.regI) * (self.X[u] - self.Y[i])
                self.Bu[u] += self.lRate * (error - self.regB * bu)
                self.Bi[i] += self.lRate * (error - self.regB * bi)
            iteration += 1

After Change

                user, item, rating = entry
                error = rating - self.predict(user,item)
                u = self.dao.user[user]
                i = self.dao.item[item]
                self.loss += error ** 2
                self.loss += self.regU * (self.X[u] - self.Y[i]).dot(self.X[u] - self.Y[i])
                bu = self.Bu[u]
                bi = self.Bi[i]

Instances