scikit-learn - 运行混淆矩阵时样本数量不一致

Question

在发布此之前，我已经阅读了类似的问题，但仍然无法解决我的问题。

建立模型后，我运行model.Predict()以预测我的数据。

这是代码

y_pred = model.predict(test_generator)

它工作正常。

但是当我想知道混淆矩阵的结果时，问题就出现了。使用这样的代码：

from sklearn.metrics import confusion_matrix, precision_score, recall_score, f1_score# Confusion matrix 
confusion_matrix(test_generator, y_pred) 

precision_score(test_generator, y_pred) 

recall_score(test_generator, y_pred) 

f1_score(test_generator,y_pred) 

我有这样的错误

它说输入变量的数字不一致。有没有办法解决这个问题？

这是我的完整代码：https ://colab.research.google.com/drive/1VifdG9GXdi7-xFdx7eATIue25ZDPfPY-?usp=sharing

标签： scikit-learndeep-learningpredictiontraining-dataconfusion-matrix