9fbde21e9d
- Rework the precision metric for both CPU and GPU. - Mention it in the document. - Cleanup old support code for GPU ranking metric. - Deterministic GPU implementation. * Drop support for classification. * type. * use batch shape. * lint. * cpu build. * cpu build. * lint. * Tests. * Fix. * Cleanup error message.
70 lines
2.1 KiB
C++
70 lines
2.1 KiB
C++
/*!
|
|
* Copyright 2018-2022 by Contributors
|
|
* \file metric_common.h
|
|
*/
|
|
#ifndef XGBOOST_METRIC_METRIC_COMMON_H_
|
|
#define XGBOOST_METRIC_METRIC_COMMON_H_
|
|
|
|
#include <limits>
|
|
#include <memory> // shared_ptr
|
|
#include <string>
|
|
|
|
#include "../collective/aggregator.h"
|
|
#include "../collective/communicator-inl.h"
|
|
#include "../common/common.h"
|
|
#include "xgboost/metric.h"
|
|
|
|
namespace xgboost {
|
|
struct Context;
|
|
// Metric that doesn't need to cache anything based on input data.
|
|
class MetricNoCache : public Metric {
|
|
public:
|
|
virtual double Eval(HostDeviceVector<float> const &predts, MetaInfo const &info) = 0;
|
|
|
|
double Evaluate(HostDeviceVector<float> const &predts, std::shared_ptr<DMatrix> p_fmat) final {
|
|
double result{0.0};
|
|
auto const &info = p_fmat->Info();
|
|
collective::ApplyWithLabels(info, &result, sizeof(double),
|
|
[&] { result = this->Eval(predts, info); });
|
|
return result;
|
|
}
|
|
};
|
|
|
|
namespace metric {
|
|
// Ranking config to be used on device and host
|
|
struct EvalRankConfig {
|
|
public:
|
|
// Parsed from metric name, the top-n number of instances within a group after
|
|
// ranking to use for evaluation.
|
|
unsigned topn{std::numeric_limits<unsigned>::max()};
|
|
std::string name;
|
|
bool minus{false};
|
|
};
|
|
|
|
class PackedReduceResult {
|
|
double residue_sum_{0};
|
|
double weights_sum_{0};
|
|
|
|
public:
|
|
XGBOOST_DEVICE PackedReduceResult() {} // NOLINT
|
|
XGBOOST_DEVICE PackedReduceResult(double residue, double weight)
|
|
: residue_sum_{residue}, weights_sum_{weight} {}
|
|
|
|
XGBOOST_DEVICE
|
|
PackedReduceResult operator+(PackedReduceResult const &other) const {
|
|
return PackedReduceResult{residue_sum_ + other.residue_sum_, weights_sum_ + other.weights_sum_};
|
|
}
|
|
PackedReduceResult &operator+=(PackedReduceResult const &other) {
|
|
this->residue_sum_ += other.residue_sum_;
|
|
this->weights_sum_ += other.weights_sum_;
|
|
return *this;
|
|
}
|
|
[[nodiscard]] double Residue() const { return residue_sum_; }
|
|
[[nodiscard]] double Weights() const { return weights_sum_; }
|
|
};
|
|
|
|
} // namespace metric
|
|
} // namespace xgboost
|
|
|
|
#endif // XGBOOST_METRIC_METRIC_COMMON_H_
|