Use DART tree weights when computing SHAPs (#5050)

This PR fixes tree weights in dart being ignored when computing contributions.

* Fix ellpack page source link.
* Add tree weights to compute contribution.
This commit is contained in:
Kodi Arfer
2019-12-03 06:55:53 -05:00
committed by Jiaming Yuan
parent 64f4361b47
commit f2277e7106
12 changed files with 88 additions and 16 deletions

View File

@@ -10,6 +10,8 @@ namespace xgboost {
class EllpackPageImpl {};
EllpackPage::EllpackPage() = default;
EllpackPage::EllpackPage(DMatrix* dmat, const BatchParam& param) {
LOG(FATAL) << "Internal Error: XGBoost is not compiled with CUDA but EllpackPage is required";
}

View File

@@ -3,8 +3,8 @@
*/
#ifndef XGBOOST_USE_CUDA
#include <xgboost/data.h>
#include "ellpack_page_source.h"
namespace xgboost {
namespace data {

View File

@@ -1,14 +1,14 @@
/*!
* Copyright 2019 XGBoost contributors
*/
#include "ellpack_page_source.h"
#include <memory>
#include <utility>
#include <vector>
#include "../common/hist_util.h"
#include "ellpack_page_source.h"
#include "sparse_page_source.h"
#include "ellpack_page.cuh"
namespace xgboost {

View File

@@ -9,7 +9,6 @@
#include <memory>
#include <string>
#include "sparse_page_source.h"
#include "../common/timer.h"
namespace xgboost {