要传递给 model.compile() 的损失函数的缩写?

Joh*_*n S 2 python keras tensorflow loss-function

这里描述了不同的损失函数,但可以使用缩写而不是导入和传递实际的损失对象(请参阅此处的示例)

MeanSquaredError可以作为字符串mse, MeanAbsoluteErroras传递mae。有些是显而易见的,但其他损失如CategoricalCrossentropy、CosineSimilarity、 或 呢Hinge?

Kav*_*veh 6

以下是一些损失函数及其字符串别名(所有这些都可以从 导入tf.keras.losses):

班级名称 字符串别名
均方误差() mean_squared_error, mse,MSE
平均绝对误差() mean_absolute_error, mae,MAE
平均绝对百分比误差() mean_absolute_percentage_error, mape,MAPE
均方对数误差() mean_squared_logarithmic_error, msle,MSLE
KL散度() kl_divergence, kullback_leibler_divergence, kld,KLD
合页() hinge
余弦相似度() cosine_similarity
LogCosh() log_cosh,logcosh
分类交叉熵() categorical_crossentropy
稀疏分类交叉熵() sparse_categorical_crossentropy
二元交叉熵() binary_crossentropy

正如您所看到的,其中一些有多个字符串别名,而大多数只有一个别名。

参考:keras.losses源码