LSTM nlp多类模型中的错误:-ValueError:形状(无,1)和(无,3)不兼容

时间:2020-10-17 06:49:45

标签: python tensorflow lstm valueerror

我正在研究一个电子邮件分类器(使用Tensorflow和keras),我自己为此创建了一个数据集。它具有3个类-spamfeedback/advicecomplaint。我拟合模型时出现错误:ValueError: Shapes (None, 1) and (None, 3) are incompatible。我读了一些答案,说我需要在训练和验证数据to_categorical()上使用(X_train, X_val, Y_train, Y_val)。但这还会带来新的错误,例如轴错误,更常见的是我关于colab崩溃的会话。我完全感到困惑,请帮忙:

我的代码: https://colab.research.google.com/drive/1GWMx1p9XhQjbC-CD5CQsVpVIn4jvxG6m?usp=sharing

我的数据集: https://www.kaggle.com/ayaanmustafa/email-classification

1 个答案:

答案 0 :(得分:0)

请参阅我的答案here。简而言之,将目标变量转换为一个热点。它的形状是(batch_size, 1),应该是(batch_size, 3)