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

[LEARNER] refactor learner

Browse Source
This commit is contained in:
tqchen
2016-01-04 01:31:44 -08:00
parent 4b4b36d047
commit 0d95e863c9
14 changed files with 470 additions and 517 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/objective/multiclass_obj.cc
View File

@@ -30,7 +30,7 @@ class SoftmaxMultiClassObj : public ObjFunction {
explicit SoftmaxMultiClassObj(bool output_prob)
: output_prob_(output_prob) {
}
void Init(const std::vector<std::pair<std::string, std::string> >& args) override {
void Configure(const std::vector<std::pair<std::string, std::string> >& args) override {
param_.InitAllowUnknown(args);
}
void GetGradient(const std::vector<float>& preds,

2
src/objective/rank_obj.cc
View File

@@ -32,7 +32,7 @@ struct LambdaRankParam : public dmlc::Parameter<LambdaRankParam> {
// objective for lambda rank
class LambdaRankObj : public ObjFunction {
public:
void Init(const std::vector<std::pair<std::string, std::string> >& args) override {
void Configure(const std::vector<std::pair<std::string, std::string> >& args) override {
param_.InitAllowUnknown(args);
}
void GetGradient(const std::vector<float>& preds,

4
src/objective/regression_obj.cc
View File

@@ -76,7 +76,7 @@ struct RegLossParam : public dmlc::Parameter<RegLossParam> {
template<typename Loss>
class RegLossObj : public ObjFunction {
public:
void Init(const std::vector<std::pair<std::string, std::string> >& args) override {
void Configure(const std::vector<std::pair<std::string, std::string> >& args) override {
param_.InitAllowUnknown(args);
}
void GetGradient(const std::vector<float> &preds,
@@ -155,7 +155,7 @@ struct PoissonRegressionParam : public dmlc::Parameter<PoissonRegressionParam> {
class PoissonRegression : public ObjFunction {
public:
// declare functions
void Init(const std::vector<std::pair<std::string, std::string> >& args) override {
void Configure(const std::vector<std::pair<std::string, std::string> >& args) override {
param_.InitAllowUnknown(args);
}