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

small fix

Browse Source
This commit is contained in:
kalenhaha 2014-04-06 22:54:41 +08:00
parent 6bc71df494
commit 1fa367b220
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dev/README Normal file
View File

@ -0,0 +1 @@
This directory contains codes under development

4
dev/rank/xgboost_rank_main.cpp
View File

@ -17,12 +17,12 @@ int main(int argc, char *argv[]) {
xgboost::random::Seed(0); xgboost::random::Seed(0);
xgboost::base::BoostTask tsk; xgboost::base::BoostTask tsk;
xgboost::utils::ConfigIterator itr(argv[1]); xgboost::utils::ConfigIterator itr(argv[1]);
int learner_index = 0; /* int learner_index = 0;
while (itr.Next()){ while (itr.Next()){
if (!strcmp(itr.name(), "learning_task")){ if (!strcmp(itr.name(), "learning_task")){
learner_index = atoi(itr.val()); learner_index = atoi(itr.val());
} }
} }*/
xgboost::rank::RankBoostLearner* rank_learner = new xgboost::rank::RankBoostLearner; xgboost::rank::RankBoostLearner* rank_learner = new xgboost::rank::RankBoostLearner;
xgboost::base::BoostLearner *parent = static_cast<xgboost::base::BoostLearner*>(rank_learner); xgboost::base::BoostLearner *parent = static_cast<xgboost::base::BoostLearner*>(rank_learner);
tsk.SetLearner(parent); tsk.SetLearner(parent);