【机器学习】 From version 0.21, test_size will always complement train_size unless both are specified

最新推荐文章于 2022-09-01 23:09:39 发布

aoxiang_ywj

最新推荐文章于 2022-09-01 23:09:39 发布

阅读量3k

点赞数

分类专栏：机器学习

本文链接：https://blog.csdn.net/baidu_37973494/article/details/93136681

版权

机器学习专栏收录该内容

3 篇文章 0 订阅

订阅专栏

x_train, x_valid, y_train, y_valid = train_test_split(x, y, train_size=0.8, random_state=1)

错误提示：
/anaconda3/envs/mlenvment/lib/python3.7/site-packages/sklearn/model_selection/_split.py:2069: FutureWarning: From version 0.21, test_size will always complement train_size unless both are specified.
FutureWarning)

解决办法：加入test_size参数 x_train,x_test,y_train,y_test = train_test_split(X,Y,train_size= 0.8,test_size=0.2,random_state=1)

修改后为：x_train, x_valid, y_train, y_valid = train_test_split(x, y, train_size=0.8, test_size=0.2, random_state=1)