我正在尝试使用 scikit-learn 的 MLPClassifier 训练 MLP。
from sklearn.neural_network import MLPClassifier
我正在用 5400 次迭代训练 mlp,但这大约需要 40 分钟。我做错了什么?这是创建的 mlp:
mlp= MLPClassifier(hidden_layer_sizes=(128),activation='relu',solver='adam',batch_size=500,shuffle=False,verbose=True)
这是我的代码的培训部分:
for j in range (5400):
mlp.partial_fit(train_X, y_train,classes=np.unique(y_train))#1 step
train_X 维度为 (27000,784),即 27000 个样本,每个样本为 28*28=784 像素。
我的处理器是 Intel i7-9750H RAM 大小是 16GB。