sklearn学习
给定数据集X和类别标签y,将数据集按一定比例随机切分为训练集和测试集。
代码
"""
功能:数据集按比例切分为训练集和测试集
时间:2017年3月11日 12:48:57
"""
from sklearn.cross_validation import train_test_split
X = [[u"这是", u"第1个", u"测试"]] * 100 + [[u"这是", u"第2个", u"测试"]] * 100
y = [1] * 100 + [2] * 100
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
print len(X_train), len(X_test)
for i in range(len(X_test)):
print "".join(X_test[i]), y_test[i]
if __name__ == "__main__":
pass
实验结果
切分后的训练集和测试集标签仍然一一对应。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)