sketch of xgboost-spark

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

View File

@@ -4,7 +4,7 @@
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
<groupId>org.dmlc</groupId>
<groupId>ml.dmlc</groupId>
<artifactId>xgboostjvm</artifactId>
<version>0.1</version>
</parent>
@@ -13,7 +13,7 @@
<packaging>jar</packaging>
<dependencies>
<dependency>
<groupId>org.dmlc</groupId>
<groupId>ml.dmlc</groupId>
<artifactId>xgboost4j</artifactId>
<version>0.1</version>
</dependency>

View File

@@ -49,6 +49,7 @@ public class CrossValidation {
//set additional eval_metrics
String[] metrics = null;
String[] evalHist = XGBoost.crossValiation(params, trainMat, round, nfold, metrics, null, null);
String[] evalHist = XGBoost.crossValidation(params, trainMat, round, nfold, metrics, null,
null);
}
}