|
|
@@ -92,12 +92,12 @@ if __name__ == "__main__":
|
|
|
score_path = "./models/recall/itemCF/matrix/score.csv"
|
|
|
similarity_path = "./models/recall/itemCF/matrix/similarity.csv"
|
|
|
itemcf_model = ItemCF(score_path, similarity_path)
|
|
|
- # itemcf_model.train(n_jobs=4)
|
|
|
- # recommend_list = itemcf_model.inference(110102)
|
|
|
+ itemcf_model.train(n_jobs=4)
|
|
|
+ recommend_list = itemcf_model.inference(110102)
|
|
|
# itemcf_model.to_redis_zset()
|
|
|
# print(len(recommend_list))
|
|
|
# print(recommend_list)
|
|
|
- joblib.dump(itemcf_model, "itemCF.model")
|
|
|
+ # joblib.dump(itemcf_model, "itemCF.model")
|
|
|
|
|
|
# model = joblib.load("./itemCF.model")
|
|
|
# recommend_list = model.inference(110102)
|