我正在研究一个电子邮件分类器(使用Tensorflow和keras),我自己为此创建了一个数据集。它具有3个类-spam
,feedback/advice
和complaint
。我拟合模型时出现错误: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