凯拉斯:如果我多次训练10个时代,是否需要重新加载模型?

2024-04-25 19:43:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在训练一个模型,想使用地图度量。由于某些原因,tensorflow mean\ u average\ u precision\ u at\ u k对我不起作用,但sklearn average\ u precision\得分起作用。 如何访问keras的模型输出以执行sklearn度量? 我是否可以编译一次模型并适合10个时代,执行度量并再次适合10个时代?还是每次都需要保存模型并重新加载? 谢谢


Tags: 模型度量tensorflow地图原因sklearnmeanat
2条回答

对于任何一个可能有同样问题的人。似乎在TensorFlow1.14中,Keras的实现保持了模型的权重,但重新启动了优化器,这会导致在多次重复.fit()函数时出现不好的结果。每次使用.fit()时,我的损失约为800,每次使用5个历元时,我的损失约为2800。你知道吗

Can I compile the model one time and fit for 10 epochs, perform the metric and fit again for 10 epochs

是的,当然。你知道吗

该模型将保持调用fit()之间的训练权重。你想叫多少次都行。你知道吗

相关问题 更多 >