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'])






网友评论