当输入形状不同时，keras 中的迁移学习

Question

我有一个关于迁移学习的问题。假设有一个神经网络模型接受形状 (250,7) 的输入。我想用这个预训练模型的权重初始化模型，然后在我的数据集上训练它以根据我的数据集更新权重。但是我的数据集是形状（251,8）。考虑到我的输入形状不同，有没有办法使用预训练模型初始化权重？如果是这样，我该怎么做？我们将不胜感激。

Answer 1

您可以尝试在迁移学习模型之前添加另一层。就像最后一层一样，这将更新它在您的数据集上的权重并且应该可以正常工作。