Fix inplace prediction interval. (#6259)

* Add back the interval in call.
* Make the interval non-optional.
This commit is contained in:
Jiaming Yuan
2020-10-28 13:13:59 +08:00
committed by GitHub
parent cc76724762
commit 3310e208fd
6 changed files with 17 additions and 12 deletions

View File

@@ -133,7 +133,7 @@ class Learner : public Model, public Configurable, public dmlc::Serializable {
virtual void InplacePredict(dmlc::any const& x, std::string const& type,
float missing,
HostDeviceVector<bst_float> **out_preds,
uint32_t layer_begin = 0, uint32_t layer_end = 0) = 0;
uint32_t layer_begin, uint32_t layer_end) = 0;
void LoadModel(Json const& in) override = 0;
void SaveModel(Json* out) const override = 0;