Sklearn 模块缺失并且不确定使用什么来代替？

Question

我目前正在遵循本指南

https://www.geeksforgeeks.org/rainfall-prediction-using-machine-learning-python/

但是我在以下代码行中收到此错误。

for i in range(3):

    models[i].fit(X, Y)


    print(f'{models[i]} : ')


    train_preds = models[i].predict_proba(X) 

    print('Training Accuracy : ', metrics.roc_auc_score(Y, train_preds[:,1]))


    val_preds = models[i].predict_proba(X_val) 

    print('Validation Accuracy : ', metrics.roc_auc_score(Y_val, val_preds[:,1]))

    print()



    metrics.plot_confusion_matrix(models[2], X_val, Y_val)

    plt.show()

我收到此错误“AttributeError：模块‘sklearn.metrics’没有属性‘plot_confusion_matrix’。您的意思是：‘pair_confusion_matrix’吗？”

显然绘图混淆矩阵已被弃用，但我不确定现在该怎么做，因为我确实需要这张图。

希望你能帮忙！！

已尝试查看文档，但找不到替换此旧功能的内容。

Answer 1

您应该尝试使用此处提到的

sklearn.metrics.ConfusionMatrixDisplay

：ImportError：无法从“sklearn.metrics”导入名称“plot_confusion_matrix”