首页 > 解决方案 > 在 Keras 模型中添加嵌入层以进行 NLP 分类时出现形状错误

问题描述

我有 4 个不同类别的文本分类问题。我已将这些类转换为 one-hot 表示。这就是我处理 X 数据的方式: 在此处输入图像描述

在此处输入图像描述

使用这些参数我得到以下错误:

ValueError: Shapes (None, 4) and (None, 500, 4) are incompatible

标签: pythonkerasdeep-learningnlp

解决方案


您需要在 Embedding Layer 中指定等于 max_length 或 x_train_padded.shape[1] 的 input_shape 参数,以便它可以以特征表示的方式通过 Dense Layer


推荐阅读