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 的包含依赖图

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

前往此文件的源代码。

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

命名空间
	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);
  });