要编译 tf.keras 模型,需要:
model.compile(
optimizer='sgd',#'adam', or what not
loss='sparse_categorical_crossentropy',
metrics=['accuracy','mae']
)
还有哪些其他指标可以用作“metrics”的参数值?我试图在文档中寻找它们,但没有成功。我正在寻找整个列表。
编辑:我知道可以使用明确的形式。例如
metrics=[tf.keras.metrics.MeanAbsoluteError()] #instead of "mae"
但短标签稍后对于绘图更有用:
lossplot = df.plot(y="mae", title="Loss vs Epochs", legend=False)
lossplot.set(xLabel="Epochs", yLabel="Loss")
尝试使用
Metrics
代替这里:https://www.tensorflow.org/api_docs/python/tf/keras/metrics
例如:
model.compile(optimizer='rmsprop',
loss='categorical_crossentropy',
metrics=[tf.keras.metrics.Recall()])
来自文档:
当您传递字符串“accuracy”或“acc”时,我们将其转换为一 tf.keras.metrics.BinaryAccuracy, tf.keras.metrics.CategoricalAccuracy, tf.keras.metrics.SparseCategoricalAccuracy 基于形状 目标和模型输出。
并且没有关于所有转换的列表。