LSTM nlp多类模型中出错：ValueError:形状（无，1）和（无，3）不兼容

Question

我正在开发一个电子邮件分类器（使用Tensorflow和keras），我自己为其创建了一个数据集。它有3个类-spam、feedback/advice和complaint。它给出了一个错误：ValueError: Shapes (None, 1) and (None, 3) are incompatible当我适合这个模型时。我读了一些关于它的答案，上面说我需要在我的培训和验证数据上使用to_categorical()。但这也带来了一个新的错误，比如axis错误，更常见的是我在colab上的会话崩溃。我完全糊涂了，请帮忙：

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

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

Answer 1

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