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',
    ]
)
上一篇:Docker:我的镜像去哪了


下一篇:评价指标(metrics)