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

fix base score, and print message

Browse Source
This commit is contained in:
tqchen@graphlab.com
2014-08-18 10:53:15 -07:00
parent 04e04ec5a0
commit f6c763a2a7
7 changed files with 39 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/learner/objective-inl.hpp
View File

@@ -124,7 +124,7 @@ class RegLossObj : public IObjFunction{
loss.SecondOrderGradient(p, info.labels[j]) * w);
}
}
virtual const char* DefaultEvalMetric(void) {
virtual const char* DefaultEvalMetric(void) const {
return loss.DefaultEvalMetric();
}
virtual void PredTransform(std::vector<float> *io_preds) {
@@ -135,6 +135,9 @@ class RegLossObj : public IObjFunction{
preds[j] = loss.PredTransform(preds[j]);
}
}
virtual float ProbToMargin(float base_score) const {
return loss.ProbToMargin(base_score);
}
protected:
float scale_pos_weight;
@@ -192,7 +195,7 @@ class SoftmaxMultiClassObj : public IObjFunction {
virtual void EvalTransform(std::vector<float> *io_preds) {
this->Transform(io_preds, 0);
}
virtual const char* DefaultEvalMetric(void) {
virtual const char* DefaultEvalMetric(void) const {
return "merror";
}
@@ -320,7 +323,7 @@ class LambdaRankObj : public IObjFunction {
}
}
}
virtual const char* DefaultEvalMetric(void) {
virtual const char* DefaultEvalMetric(void) const {
return "map";
}