【发布时间】:2017-11-24 18:38:33
【问题描述】:
我正在学习如何使用 scikit-learn 开发反向传播神经网络。我仍然对如何在我的神经网络中实现 k 折交叉验证感到困惑。我希望你们能帮助我。我的代码如下:
import numpy as np
from sklearn.model_selection import KFold
from sklearn.neural_network import MLPClassifier
f = open("seeds_dataset.txt")
data = np.loadtxt(f)
X=data[:,0:]
y=data[:,-1]
kf = KFold(n_splits=10)
X_train, X_test, y_train, y_test = X[train], X[test], y[train], y[test]
clf = MLPClassifier(solver='lbfgs', alpha=1e-5, hidden_layer_sizes=(5, 2), random_state=1)
clf.fit(X, y)
MLPClassifier(activation='relu', alpha=1e-05, batch_size='auto',
beta_1=0.9, beta_2=0.999, early_stopping=False,
epsilon=1e-08, hidden_layer_sizes=(5, 2), learning_rate='constant',
learning_rate_init=0.001, max_iter=200, momentum=0.9,
nesterovs_momentum=True, power_t=0.5, random_state=1, shuffle=True,
solver='lbfgs', tol=0.0001, validation_fraction=0.1, verbose=False,
warm_start=False)
【问题讨论】:
-
我认为您的意图是您想要进行 kfold 拆分以及评分。你可以使用
cross_val_score。它会为你做 Kfold,同时生成分数列表。
标签: python scikit-learn neural-network