【发布时间】:2020-05-24 07:56:12
【问题描述】:
我已经在不同的数据集上训练了两个模型,并将每个模型的权重保存为 ModelA.h5 和 ModelB.h5 我想平均这些权重并创建一个名为 ModelC.h5 的新文件夹并将其加载到相同的模型架构上。 我该怎么做?
【问题讨论】:
标签: keras deep-learning conv-neural-network keras-layer
我已经在不同的数据集上训练了两个模型,并将每个模型的权重保存为 ModelA.h5 和 ModelB.h5 我想平均这些权重并创建一个名为 ModelC.h5 的新文件夹并将其加载到相同的模型架构上。 我该怎么做?
【问题讨论】:
标签: keras deep-learning conv-neural-network keras-layer
不能像这样添加在不同数据集上训练的模型。它看起来像这样。比方说,训练一个人将 1000 张图像分类为 5 个类别,然后训练另一个人将另外 1000 张图像分类为相同的 5 个类别。现在,您想将它们合二为一。
相反,您可以做的是综合这两个网络。有多种方法可以使用 Max Voting、Averaging 或 Weighted Average、Bagging 和 Boosting 等方法来集成两种模型的预测。Ensemble 有助于将弱分类器提升为一个强分类器。
您可以参考此链接以了解有关不同类型合奏的更多信息:Link
【讨论】: