sklearn中随机分割训练集和测试集

from sklearn.model_selection import train_test_split
X_train,X_test,Y_train,Y_test = train_test_split(X,Y,test_size=0.3,random_state=0)

这里的random_state就是为了保证程序每次运行都分割一样的训练集和测试集。否则,同样的算法模型在不同的训练集和测试集上的效果不一样。

猜你喜欢

转载自blog.csdn.net/banxiaCSDN/article/details/112957133
今日推荐