From 6ddc5d0f412e5ab7dc81786421d11f399203d011 Mon Sep 17 00:00:00 2001 From: jiangzhonglian Date: Mon, 10 Jul 2017 12:37:57 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=8E=A8=E8=8D=90=E7=B3=BB?= =?UTF-8?q?=E7=BB=9F=E5=86=85=E5=AE=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/python/16.RecommenderSystems/RS-sklearn-rating.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/python/16.RecommenderSystems/RS-sklearn-rating.py b/src/python/16.RecommenderSystems/RS-sklearn-rating.py index f00200fc..768f06c7 100644 --- 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