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

refactor config

Browse Source
This commit is contained in:
tqchen
2014-08-15 21:02:33 -07:00
parent dafa44753a
commit 3589e8252f
3 changed files with 189 additions and 213 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
learner/objective.h
View File

@@ -57,7 +57,6 @@ class IObjFunction{
return base_score;
}
};
} // namespace learner
} // namespace xgboost
@@ -68,10 +67,10 @@ namespace xgboost {
namespace learner {
/*! \brief factory funciton to create objective function by name */
inline IObjFunction* CreateObjFunction(const char *name) {
if (!strcmp("reg:linear", name)) return new RegLossObj( LossType::kLinearSquare );
if (!strcmp("reg:logistic", name)) return new RegLossObj( LossType::kLogisticNeglik );
if (!strcmp("binary:logistic", name)) return new RegLossObj( LossType::kLogisticClassify );
if (!strcmp("binary:logitraw", name)) return new RegLossObj( LossType::kLogisticRaw );
if (!strcmp("reg:linear", name)) return new RegLossObj(LossType::kLinearSquare);
if (!strcmp("reg:logistic", name)) return new RegLossObj(LossType::kLogisticNeglik);
if (!strcmp("binary:logistic", name)) return new RegLossObj(LossType::kLogisticClassify);
if (!strcmp("binary:logitraw", name)) return new RegLossObj(LossType::kLogisticRaw);
utils::Error("unknown objective function type: %s", name);
return NULL;
}