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

add AMS metric

Browse Source
This commit is contained in:
tqchen 2014-05-14 11:30:45 -07:00
parent 121348c0d7
commit ae9d937510
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
regrank/xgboost_regrank_eval.h
View File

@ -99,6 +99,27 @@ namespace xgboost{
}
};
/*! \brief AMS */
struct EvalAMS : public IEvaluator{
virtual float Eval(const std::vector<float> &preds,
const DMatrix::Info &info) const {
const unsigned ndata = static_cast<unsigned>(preds.size());
double s_tp = 0.0, b_fp = 0.0;
#pragma omp parallel for reduction(+:s_tp,b_fp) schedule( static )
for (unsigned i = 0; i < ndata; ++i){
const float wt = info.GetWeight(i);
if (preds[i] > 0.5f){
if( info.labels[i] > 0.5f ) s_tp += wt;
else b_fp += wt;
}
}
const double br = 10.0;
return sqrtf( 2*((s_tp+b_fp+br) * log( 1.0 + s_tp/(b_fp+br) ) - s_tp) );
}
virtual const char *Name(void) const{
return "ams";
}
};
/*! \brief Error */
struct EvalMatchError : public IEvaluator{
@ -316,6 +337,7 @@ namespace xgboost{
if (!strcmp(name, "error")) evals_.push_back(new EvalError());
if (!strcmp(name, "merror")) evals_.push_back(new EvalMatchError("merror"));
if (!strcmp(name, "mabserror")) evals_.push_back(new EvalMatchError("mabserror"));
if (!strcmp(name, "ams")) evals_.push_back(new EvalAMS());
if (!strcmp(name, "logloss")) evals_.push_back(new EvalLogLoss());
if (!strcmp(name, "auc")) evals_.push_back(new EvalAuc());
if (!strncmp(name, "pre@", 4)) evals_.push_back(new EvalPrecision(name));