lightgbm.record_evaluation

lightgbm.record_evaluation(eval_result)[源代码]

创建一个回调,将评估历史记录到 eval_result 中。

参数:

eval_result (dict) –

用于存储所有验证集的所有评估结果的字典。应在调用 record_evaluation() 之外初始化此字典,且应为空。字典中的任何初始内容都将被删除。

示例

假设有两个验证集分别名为 ‘eval’ 和 ‘train’,以及一个评估指标名为 ‘logloss’,则模型训练完成后,此字典将具有以下结构:

{
 'train':
     {
      'logloss': [0.48253, 0.35953, ...]
     },
 'eval':
     {
      'logloss': [0.480385, 0.357756, ...]
     }
}

返回:

callback – 将评估历史记录到传入字典中的回调。

返回类型:

_RecordEvaluationCallback