metric.h 文件参考

xgboost 中支持的评估指标函数的接口。 更多...

#include <dmlc/registry.h>
#include <xgboost/base.h>
#include <xgboost/data.h>
#include <xgboost/host_device_vector.h>
#include <xgboost/model.h>
#include <functional>
#include <memory>
#include <string>
#include <utility>
#include <vector>

metric.h 的包含依赖图

此图显示哪些文件直接或间接包含此文件

转到此文件的源代码。

类
类	xgboost::Metric
	用于评估模型性能的评估指标接口。这与训练无关，仅用于评估目的。更多...
结构	xgboost::MetricReg
	用于 Metric 工厂函数的注册表条目。附加参数 const char* param 给出 @ 后的值，可以为 null。例如，metric map@3，则：param == "3"。 更多...

命名空间
	xgboost
	Learner 接口，将目标、gbm 和评估整合在一起。这是用户界面 XGBoost 训练模块。

宏
#define	XGBOOST_REGISTER_METRIC(UniqueId, Name)
	注册指标的宏。 更多...

详细描述

xgboost 中支持的评估指标函数的接口。

版权所有 2014-2023 XGBoost 贡献者

作者

陈天奇，陈凯龙

宏定义文档

XGBOOST_REGISTER_METRIC

#define XGBOOST_REGISTER_METRIC	(		UniqueId,
			Name
	)

值

  ::xgboost::MetricReg& __make_ ## MetricReg ## _ ## UniqueId ## __ = \
::dmlc::Registry< ::xgboost::MetricReg>::Get()->__REGISTER__(Name)

注册指标的宏。

// 注册目标 ndcg@k 的示例
XGBOOST_REGISTER_METRIC(RMSE, "ndcg")
.describe("均方根误差。")
.set_body([](const char* param) {
    int at_k = atoi(param);
    return new NDCG(at_k);
  });