如何继续使用新的类和数据训练模型?

时间:2018-05-12 10:36:33

标签: python-3.x tensorflow machine-learning rnn

我已经成功训练了一个模型,现在我想继续用新数据训练它。如果具有相同数量的类的给定数据,它可以正常工作。但是最初有更多数据会给我错误:

  

ValueError:形状(?,14)和(?,21)不兼容

如何动态增加训练模型中的类数或如何使模型接受较少数量的类?我是否需要将这些类保存在pickle文件中?

1 个答案:

答案 0 :(得分:0)

  • 最好的办法是从头开始训练您的网络,并将输出图层调整为新的输出类大小。
  • 如果再培训是一个问题,那么保持训练有素的网络不变,只丢弃最后一层。添加一个具有适当输出大小的新图层,初始化为随机权重,然后对整个网络进行微调(训练)。