Linux 拨号vps windows公众号手机端

Keras中怎么定义损失函数和优化器

lewis 5年前 (2020-04-17) 阅读数 9 #大数据
文章标签 keras

在Keras中,可以通过compile方法来定义损失函数和优化器。例如:

model.compile(loss='mean_squared_error', optimizer='adam')

在上面的例子中,我们使用了均方误差作为损失函数,使用了Adam优化器。当然,你也可以自定义损失函数和优化器,例如:

from keras import optimizers
from keras import losses

model.compile(loss=losses.mean_squared_error, optimizer=optimizers.Adam())

这样就可以自定义损失函数和优化器了。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门