add cv for python

This commit is contained in:
tqchen
2014-09-03 22:43:55 -07:00
parent 586d6ae740
commit da9c856701
6 changed files with 91 additions and 10 deletions

View File

@@ -80,6 +80,9 @@ class EvalSet{
}
return result;
}
inline size_t Size(void) const {
return evals_.size();
}
private:
std::vector<const IEvaluator*> evals_;

View File

@@ -244,7 +244,9 @@ class BoostLearner {
obj_->SetParam(cfg_[i].first.c_str(), cfg_[i].second.c_str());
gbm_->SetParam(cfg_[i].first.c_str(), cfg_[i].second.c_str());
}
evaluator_.AddEval(obj_->DefaultEvalMetric());
if (evaluator_.Size() == 0) {
evaluator_.AddEval(obj_->DefaultEvalMetric());
}
}
/*!
* \brief get un-transformed prediction