cross-validation - 无法获得训练集和测试集
问题描述
我应用 k 折交叉验证将数据拆分为训练集和测试集。但是当我想获得训练和测试集时,我遇到了这些错误:
AttributeError:“numpy.ndarray”对象没有属性“iloc”
谢谢你的帮助。
y = df_dummies['Churn'].values
X = df_dummies.drop(columns = ['Churn'])
from sklearn.preprocessing import MinMaxScaler
features = X.columns.values
scaler = MinMaxScaler(feature_range = (0,1))
scaler.fit(X)
X = pd.DataFrame(scaler.transform(X))
X.columns = features
from sklearn.model_selection import KFold
kf=KFold(n_splits=5,shuffle=True)
for train,test in kf.split(X):
print("%s %s" % (train,test))
for train_index, test_index in kf.split(X):
print("TRAIN:", train_index, "TEST:", test_index)
X_train, X_test = X.iloc[train_index], X.iloc[test_index]
y_train, y_test = y.iloc[train_index], y.iloc[test_index]
from sklearn.linear_model import LogisticRegression
CLF = LogisticRegression().fit(X_train, y_train)
print('Accuracy of Logistic regression classifier on training set: {:.2f}'
.format(CLF.score(X_train, y_train)))
print('Accuracy of Logistic regression classifier on test set: {:.2f}'
.format(CLF.score(X_test, y_test)))
NameError: name 'y_train' is not defined
解决方案
问题是df_dummies['Churn'].values
返回一个数组而不是数据框。但是您正试图从不存在的数组中获取属性。iloc
函数在pandas.DataFrame
.
改为使用y = df_dummies['Churn']
。
PS:我不知道如何将这些类型的问题迁移到姐妹站点。也许,知道的人可以将此迁移到交叉验证。
推荐阅读
- reactjs - 打字稿:编译失败
- reactjs - 如何摆脱这个 - 消息:{'你没有订阅这个 API。'}。?
- php - Symfony 5 控制器不会验证错误
- javascript - 正确添加元素到 React Native 中的对象数组
- javascript - 将变量添加到 ajax 中的新 FormData 对象的问题
- nebular - 如何更改 NbSuffix/NbPrefix 的颜色
- python - Seaborn Catplot 误差与分类预测变量
- c# - UWP Webview - 选项卡标题更改时触发事件(DocumentTitle)
- pytest - 在 Windows 10 终端中使用 py.test 命令出错
- javascript - 每次修改单元格时创建/更改日期戳