更新推荐系统内容

2026-02-13 07:15:26 +08:00 · 2017-07-10 12:37:57 +08:00
parent b4d92da982
commit 6ddc5d0f41
1 changed files with 1 additions and 0 deletions
--- a/src/python/16.RecommenderSystems/RS-sklearn-rating.py
+++ b/src/python/16.RecommenderSystems/RS-sklearn-rating.py
@@ -23,6 +23,7 @@ def splitData(dataFile, test_size):

    print 'Number of users = ' + str(n_users) + ' | Number of movies = ' + str(n_items)
    train_data, test_data = cv.train_test_split(df, test_size=test_size)
+    print "数据量：", len(train_data), len(test_data)
    return df, n_users, n_items, train_data, test_data