【代码模版】有监督学习下特征/标签和训练集/测试集的划分

记录该模版就是为了方便调用,免得每次都要写代码

# 划分特征与标签
X = DataFrame.iloc[:, DataFrame.columns != 'target_name']
Y = DataFrame.iloc[:, DataFrame.columns == 'target_name']
# 划分训练集与测试集
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)

补充:对于需要选择特定特征的情况:

X = DataFrame[['column_name1', 'column_name2', ... , 'column_namen']]
Y = DataFrame.iloc[:, DataFrame.columns == 'target_name']
发布了22 篇原创文章 · 获赞 0 · 访问量 929

猜你喜欢

转载自blog.csdn.net/weixin_44680262/article/details/104608963