Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PLUGIN] Add plugin system

Browse Source
This commit is contained in:
tqchen
2016-01-12 11:56:22 -08:00
parent 36c389ac46
commit 96f4542a67
14 changed files with 457 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/objective/regression_obj.cc
View File

@@ -108,10 +108,10 @@ class RegLossObj : public ObjFunction {
LOG(FATAL) << Loss::LabelErrorMsg();
}
}
virtual const char* DefaultEvalMetric() const {
const char* DefaultEvalMetric() const override {
return Loss::DefaultEvalMetric();
}
virtual void PredTransform(std::vector<float> *io_preds) {
void PredTransform(std::vector<float> *io_preds) override {
std::vector<float> &preds = *io_preds;
const bst_omp_uint ndata = static_cast<bst_omp_uint>(preds.size());
#pragma omp parallel for schedule(static)
@@ -119,7 +119,7 @@ class RegLossObj : public ObjFunction {
preds[j] = Loss::PredTransform(preds[j]);
}
}
virtual float ProbToMargin(float base_score) const {
float ProbToMargin(float base_score) const override {
return Loss::ProbToMargin(base_score);
}