记录该模版就是为了方便调用,免得每次都要写代码
# 划分特征与标签
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']