python 将数据随机分为训练集和测试集

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24

# -*- coding: utf-8 -*- """ Created on Tue Jun 23 15:24:19 2015   @author: hd """   from sklearn import cross_validation   c = [] j=0 filename = r'C:\Users\hd\Desktop\bookmarks\bookmarks.arff' out_train = open(r'C:\Users\hd\Desktop\bookmarks\train.arff','w') out_test = open(r'C:\Users\hd\Desktop\bookmarks\test.arff','w')   for line in open(filename): #    items = line.strip().split()     c.append(line)    c_train,c_test = cross_validation.train_test_split(c,test_size = 0.6) for i in c_train:     out_train.write(i) for i in c_test:     out_test.write(i)