在scikit-learn中使用mlpclassifier按时期制作精确度图表

时间:2018-03-22 12:49:49

标签: python tensorflow plot scikit-learn epoch

我想通过scikit-learn中的时代来测试准确性。但是,到目前为止,我一直没有成功。

这是我使用mlpclassifier进行分类的代码部分:

NUM_EPOCHS = 1000
LOG_FOR_EVERY = 10

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
clf = MLPClassifier(hidden_layer_sizes=(18, 175, 256), batch_size=528,
                    learning_rate_init=0.0001, beta_1=0.001,
                    beta_2=0.001, max_iter=1, warm_start=True)


for i in range(NUM_EPOCHS):
    clf.fit(X_train, y_train.ravel())

我也用result做了一个图表,但我需要让它连续并进一步提高准确性。

为什么准确度不会增加?

1 个答案:

答案 0 :(得分:-1)

请尝试:

clf.fit(X_train, y_train.ravel(), epochs=NUM_EPOCHS)