示例#1
0
文件: Average.py 项目: kapild/kaggle
    def get_output_line(self, review_item):

        bus_rating = BaseModel.get_business_avg_stars(self.bus, self.user,
                                                      review_item,
                                                      self.test_bus,
                                                      self.test_user)

        user_rating = BaseModel.get_user_avg_stars(self.bus, self.user,
                                                   review_item, self.test_bus,
                                                   self.test_user)

        prediction = float(bus_rating + user_rating) / 2
        return prediction
示例#2
0
文件: Average.py 项目: kapild/kaggle
	def get_output_line(self, review_item):
		
		bus_rating = BaseModel.get_business_avg_stars(
			self.bus, 
			self.user, 
			review_item,
			self.test_bus, 
			self.test_user
		)

		user_rating = BaseModel.get_user_avg_stars(
			self.bus, 
			self.user, 
			review_item,
			self.test_bus, 
			self.test_user
		)


		prediction = float(bus_rating + user_rating)/2
		return prediction