xgboost/src/common/survival_util.cc
Philip Hyunsu Cho 71b0528a2f
GPU implementation of AFT survival objective and metric (#5714)
* Add interval accuracy

* De-virtualize AFT functions

* Lint

* Refactor AFT metric using GPU-CPU reducer

* Fix R build

* Fix build on Windows

* Fix copyright header

* Clang-tidy

* Fix crashing demo

* Fix typos in comment; explain GPU ID

* Remove unnecessary #include

* Add C++ test for interval accuracy

* Fix a bug in accuracy metric: use log pred

* Refactor AFT objective using GPU-CPU Transform

* Lint

* Fix lint

* Use Ninja to speed up build

* Use time, not /usr/bin/time

* Add cpu_build worker class, with concurrency = 1

* Use concurrency = 1 only for CUDA build

* concurrency = 1 for clang-tidy

* Address reviewer's feedback

* Update link to AFT paper
2020-07-17 01:18:13 -07:00

19 lines
424 B
C++

/*!
* Copyright 2019-2020 by Contributors
* \file survival_util.cc
* \brief Utility functions, useful for implementing objective and metric functions for survival
* analysis
* \author Avinash Barnwal, Hyunsu Cho and Toby Hocking
*/
#include <dmlc/registry.h>
#include "survival_util.h"
namespace xgboost {
namespace common {
DMLC_REGISTER_PARAMETER(AFTParam);
} // namespace common
} // namespace xgboost