tensorflow model.compile() 示例

model.compile()方法用于在配置训练方法时,告知训练时用的优化器、损失函数和准确率评测标准

model.compile(
    optimizer=tf.keras.optimizers.Adam(), # 优化器
    loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True), # 损失函数
    metrics=['accuracy'], # 准确率指标
)

猜你喜欢

转载自blog.csdn.net/weixin_44493841/article/details/121473934