예제 #1
0
def recommend(filepath):
	'''
	为所有用户进行推荐,并保存计算结果
	-------------------------------
	filepath: 保存推荐结果的文件路径
	-------------------------------
	return: {uid:[entryid,...],...}
	'''
	predictions = {}
	dbhelper = DBHelper()
	uids = dbhelper.get_uids()
	recommender = LogitRegRecommender()
	for uid in uids:
		print 'predict', uid
		result = recommender.recommend(uid)
		predictions[uid] = result
	evaluator = Evaluator()
	precision,recall,F1 = evaluator.evaluate(predictions, filepath)
	print precision,recall,F1