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

sketch of xgboost-spark

Browse Source 
chooseBestBooster shall be in Boosters

remove tracker.py

rename XGBoost

remove cross-validation
This commit is contained in:
CodingCat
2015-12-22 03:29:20 -06:00
parent 4568692daf
commit 1540773340
14 changed files with 187 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j/src/test/java/ml/dmlc/xgboost4j/BoosterImplTest.java
View File

@@ -130,6 +130,6 @@ public class BoosterImplTest {
//do 5-fold cross validation
int round = 2;
int nfold = 5;
String[] evalHist = XGBoost.crossValiation(param, trainMat, round, nfold, null, null, null);
String[] evalHist = XGBoost.crossValidation(param, trainMat, round, nfold, null, null, null);
}
}

2
jvm-packages/xgboost4j/src/test/scala/ml/dmlc/xgboost4j/scala/ScalaBoosterImplSuite.scala
View File

@@ -85,6 +85,6 @@ class ScalaBoosterImplSuite extends FunSuite {
"objective" -> "binary:logistic", "gamma" -> "1.0", "eval_metric" -> "error").toMap
val round = 2
val nfold = 5
XGBoost.crossValiation(params, trainMat, round, nfold, null, null, null)
XGBoost.crossValidation(params, trainMat, round, nfold, null, null, null)
}
}