无法获取混淆矩阵和交叉验证结果

2024-05-08 05:10:18 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试执行折叠交叉验证和混淆矩阵,以便使用python中的神经网络分析我的情绪分析项目的结果

我曾尝试使用交叉验证和混淆矩阵函数来查看结果并将其保存到PNG文件中,但它似乎不起作用,我也不知道为什么,因为没有显示错误。下面是我的代码

 X, Y = np.array(X), np.array(Y)
    X_train, X_val, y_train, y_val = train_test_split(X, Y, test_size=0.25, random_state=42)

    model.compile(loss='binary_crossentropy', optimizer=sgd, metrics=['accuracy'])
    history = model.fit(X_train, y_train, validation_data=(X_val, y_val), batch_size=1, nb_epoch=100)

    pred_val = getBinaryResult(model.predict_proba(np.array(X_val)))
    class_names=np.array((0,1))
    ax = plot_confusion_matrix(y_val, pred_val, classes=class_names, title='Confusion Matrix')
    fig = plt.gcf()
    fig.savefig('confusion_matrix.png')

    plt.plot(history.history['val_acc'])
    plt.title('Validation accuracy')
    plt.ylabel('accuracy')
    plt.xlabel('epoch')
    fig = plt.gcf()
    fig.savefig('validation_accuracy.png')
    # plt.show()

我希望结果是保存在根文件夹中的图形。让我困惑的是没有错误信息,这就是为什么我在这里。任何人,请帮助我


Tags: testsizemodelnpfigtrainplt矩阵