其他分享
首页 > 其他分享> > Keras 模型构建 compile函数 参数

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',
    ]
)

标签:metrics,AUC,optimizer,Keras,squared,MeanSquaredError,compile,构建
来源: https://www.cnblogs.com/amilychensblog/p/14730167.html