Keras 模型构建 compile函数 参数

Keras Metrics handbook
metrics部分有三种写法

#第一种
model.compile(
    optimizer='adam',
    loss='mean_squared_error',
    metrics=[
        metrics.MeanSquaredError(),
        metrics.AUC(),
    ]
)
#第二种(这种还可以自定义名称)
model.compile(
    optimizer='adam',
    loss='mean_squared_error',
    metrics=[
        metrics.MeanSquaredError(name='my_mse'),
        metrics.AUC(name='my_auc'),
    ]
)
#第三种
model.compile(
    optimizer='adam',
    loss='mean_squared_error',
    metrics=[
        'MeanSquaredError',
        'AUC',
    ]
)
posted @ 2021-05-04 20:10  陈玉兰  阅读(869)  评论(0编辑  收藏  举报