使用 Skear 的 LSTM 的混淆矩阵得到错误

数据挖掘 深度学习 lstm 混淆矩阵
2022-03-11 06:09:23

我想计算我的 LSTM 模型的混淆矩阵。
y_test= (17799,1)
y_Pred= (17799,1)的形状

我使用了以下代码:
from sklearn.metrics import chaos_matrix
cm = chaos_matrix(y_test, y_pred)

出现以下错误:

raise ValueError("{0} is not supported" .format(y_type))

ValueError: Continuous is not supported

both y_test and y_predict are normalized using minmaxscaler(0,1)
两个变量的前几行如下所示:
y_test={0,0,0,0.0216212}
y_predict={0.000433061,0.000433061,0.000433061,0.000450924}

请建议如何计算混淆矩阵 tp,fp,tn,fn 。

1个回答

可以为分类问题绘制混淆矩阵,其中机器学习模型(在您的情况下为 LSTM)将目标变量预测为 N 个类别之一。

您能否确认您的问题是分类问题而不是回归问题?

如果您遇到回归问题,那么您可能需要使用不同的评估指标。