model.compile中的loss函数参数输入三种类型

2020-11-15  本文已影响0人  MasterZhang

model.compile中的loss参数,可以输入三种类型,一种是损失函数名称,一种是自定义函数,还有一种是tf.keras.losses.Loss实例.写文字太麻烦,建议用预设的实例。

输入损失函数名称
model.compile(optimizer='adam',loss='binary_crossentropy',metrics=['acc'])   
输入自定义loss函数
def myloss(y_true,y_pred):
    return tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False)
model.compile(optimizer='adam',loss=myloss,metrics=['acc'])     
输入tf.keras.losses.Loss预设实例
model.compile(optimizer='adam',loss=tf.keras.losses.SparseCategoricalCrossentropy(),metrics=['acc'])
上一篇 下一篇

猜你喜欢

热点阅读