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

Fix errors when generating javadoc

Browse Source
This commit is contained in:
damiencarol 2016-01-09 15:45:13 +01:00
parent 0958fb35ae
commit 89216e239f
6 changed files with 77 additions and 70 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
java/xgboost4j/src/main/java/org/dmlc/xgboost4j/Booster.java
View File

@ -59,7 +59,7 @@ public final class Booster {
* init Booster from dMatrixs * init Booster from dMatrixs
* @param params parameters * @param params parameters
* @param dMatrixs DMatrix array * @param dMatrixs DMatrix array
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public Booster(Iterable<Entry<String, Object>> params, DMatrix[] dMatrixs) throws XGBoostError { public Booster(Iterable<Entry<String, Object>> params, DMatrix[] dMatrixs) throws XGBoostError {
init(dMatrixs); init(dMatrixs);
@ -73,7 +73,7 @@ public final class Booster {
* load model from modelPath * load model from modelPath
* @param params parameters * @param params parameters
* @param modelPath booster modelPath (model generated by booster.saveModel) * @param modelPath booster modelPath (model generated by booster.saveModel)
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public Booster(Iterable<Entry<String, Object>> params, String modelPath) throws XGBoostError { public Booster(Iterable<Entry<String, Object>> params, String modelPath) throws XGBoostError {
init(null); init(null);
@ -103,7 +103,7 @@ public final class Booster {
* set parameter * set parameter
* @param key param name * @param key param name
* @param value param value * @param value param value
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public final void setParam(String key, String value) throws XGBoostError { public final void setParam(String key, String value) throws XGBoostError {
ErrorHandle.checkCall(XgboostJNI.XGBoosterSetParam(handle, key, value)); ErrorHandle.checkCall(XgboostJNI.XGBoosterSetParam(handle, key, value));
@ -112,7 +112,7 @@ public final class Booster {
/** /**
* set parameters * set parameters
* @param params parameters key-value map * @param params parameters key-value map
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void setParams(Iterable<Entry<String, Object>> params) throws XGBoostError { public void setParams(Iterable<Entry<String, Object>> params) throws XGBoostError {
if(params!=null) { if(params!=null) {
@ -127,7 +127,7 @@ public final class Booster {
* Update (one iteration) * Update (one iteration)
* @param dtrain training data * @param dtrain training data
* @param iter current iteration number * @param iter current iteration number
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void update(DMatrix dtrain, int iter) throws XGBoostError { public void update(DMatrix dtrain, int iter) throws XGBoostError {
ErrorHandle.checkCall(XgboostJNI.XGBoosterUpdateOneIter(handle, iter, dtrain.getHandle())); ErrorHandle.checkCall(XgboostJNI.XGBoosterUpdateOneIter(handle, iter, dtrain.getHandle()));
@ -138,7 +138,7 @@ public final class Booster {
* @param dtrain training data * @param dtrain training data
* @param iter current iteration number * @param iter current iteration number
* @param obj customized objective class * @param obj customized objective class
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void update(DMatrix dtrain, int iter, IObjective obj) throws XGBoostError { public void update(DMatrix dtrain, int iter, IObjective obj) throws XGBoostError {
float[][] predicts = predict(dtrain, true); float[][] predicts = predict(dtrain, true);
@ -151,7 +151,7 @@ public final class Booster {
* @param dtrain training data * @param dtrain training data
* @param grad first order of gradient * @param grad first order of gradient
* @param hess seconde order of gradient * @param hess seconde order of gradient
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void boost(DMatrix dtrain, float[] grad, float[] hess) throws XGBoostError { public void boost(DMatrix dtrain, float[] grad, float[] hess) throws XGBoostError {
if(grad.length != hess.length) { if(grad.length != hess.length) {
@ -166,7 +166,7 @@ public final class Booster {
* @param evalNames name for eval dmatrixs, used for check results * @param evalNames name for eval dmatrixs, used for check results
* @param iter current eval iteration * @param iter current eval iteration
* @return eval information * @return eval information
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public String evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter) throws XGBoostError { public String evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter) throws XGBoostError {
long[] handles = dMatrixs2handles(evalMatrixs); long[] handles = dMatrixs2handles(evalMatrixs);
@ -177,12 +177,12 @@ public final class Booster {
/** /**
* evaluate with given customized Evaluation class * evaluate with given customized Evaluation class
* @param evalMatrixs * @param evalMatrixs evaluation matrix
* @param evalNames * @param evalNames evaluation names
* @param iter * @param iter number of interations
* @param eval * @param eval custom evaluator
* @return eval information * @return eval information
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public String evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter, IEvaluation eval) throws XGBoostError { public String evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter, IEvaluation eval) throws XGBoostError {
String evalInfo = ""; String evalInfo = "";
@ -202,7 +202,7 @@ public final class Booster {
* @param evalNames name for eval dmatrixs, used for check results * @param evalNames name for eval dmatrixs, used for check results
* @param iter current eval iteration * @param iter current eval iteration
* @return eval information * @return eval information
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public String evalSet(long[] dHandles, String[] evalNames, int iter) throws XGBoostError { public String evalSet(long[] dHandles, String[] evalNames, int iter) throws XGBoostError {
String[] evalInfo = new String[1]; String[] evalInfo = new String[1];
@ -213,11 +213,11 @@ public final class Booster {
/** /**
* evaluate with given dmatrix, similar to evalSet * evaluate with given dmatrix, similar to evalSet
* @param evalMat * @param evalMat evaluation matrix
* @param evalName * @param evalName evaluation name
* @param iter * @param iter number of iterations
* @return eval information * @return eval information
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public String eval(DMatrix evalMat, String evalName, int iter) throws XGBoostError { public String eval(DMatrix evalMat, String evalName, int iter) throws XGBoostError {
DMatrix[] evalMats = new DMatrix[] {evalMat}; DMatrix[] evalMats = new DMatrix[] {evalMat};
@ -227,10 +227,10 @@ public final class Booster {
/** /**
* base function for Predict * base function for Predict
* @param data * @param data data
* @param outPutMargin * @param outPutMargin output margin
* @param treeLimit * @param treeLimit limit number of trees
* @param predLeaf * @param predLeaf prediction minimum to keep leafs
* @return predict results * @return predict results
*/ */
private synchronized float[][] pred(DMatrix data, boolean outPutMargin, int treeLimit, boolean predLeaf) throws XGBoostError { private synchronized float[][] pred(DMatrix data, boolean outPutMargin, int treeLimit, boolean predLeaf) throws XGBoostError {
@ -259,7 +259,7 @@ public final class Booster {
* Predict with data * Predict with data
* @param data dmatrix storing the input * @param data dmatrix storing the input
* @return predict result * @return predict result
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[][] predict(DMatrix data) throws XGBoostError { public float[][] predict(DMatrix data) throws XGBoostError {
return pred(data, false, 0, false); return pred(data, false, 0, false);
@ -270,7 +270,7 @@ public final class Booster {
* @param data dmatrix storing the input * @param data dmatrix storing the input
* @param outPutMargin Whether to output the raw untransformed margin value. * @param outPutMargin Whether to output the raw untransformed margin value.
* @return predict result * @return predict result
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[][] predict(DMatrix data, boolean outPutMargin) throws XGBoostError { public float[][] predict(DMatrix data, boolean outPutMargin) throws XGBoostError {
return pred(data, outPutMargin, 0, false); return pred(data, outPutMargin, 0, false);
@ -282,7 +282,7 @@ public final class Booster {
* @param outPutMargin Whether to output the raw untransformed margin value. * @param outPutMargin Whether to output the raw untransformed margin value.
* @param treeLimit Limit number of trees in the prediction; defaults to 0 (use all trees). * @param treeLimit Limit number of trees in the prediction; defaults to 0 (use all trees).
* @return predict result * @return predict result
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[][] predict(DMatrix data, boolean outPutMargin, int treeLimit) throws XGBoostError { public float[][] predict(DMatrix data, boolean outPutMargin, int treeLimit) throws XGBoostError {
return pred(data, outPutMargin, treeLimit, false); return pred(data, outPutMargin, treeLimit, false);
@ -297,7 +297,7 @@ public final class Booster {
Note that the leaf index of a tree is unique per tree, so you may find leaf 1 Note that the leaf index of a tree is unique per tree, so you may find leaf 1
in both tree 1 and tree 0. in both tree 1 and tree 0.
* @return predict result * @return predict result
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[][] predict(DMatrix data , int treeLimit, boolean predLeaf) throws XGBoostError { public float[][] predict(DMatrix data , int treeLimit, boolean predLeaf) throws XGBoostError {
return pred(data, false, treeLimit, predLeaf); return pred(data, false, treeLimit, predLeaf);
@ -305,7 +305,7 @@ public final class Booster {
/** /**
* save model to modelPath * save model to modelPath
* @param modelPath * @param modelPath model path
*/ */
public void saveModel(String modelPath) { public void saveModel(String modelPath) {
XgboostJNI.XGBoosterSaveModel(handle, modelPath); XgboostJNI.XGBoosterSaveModel(handle, modelPath);
@ -319,7 +319,7 @@ public final class Booster {
* get the dump of the model as a string array * get the dump of the model as a string array
* @param withStats Controls whether the split statistics are output. * @param withStats Controls whether the split statistics are output.
* @return dumped model information * @return dumped model information
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public String[] getDumpInfo(boolean withStats) throws XGBoostError { public String[] getDumpInfo(boolean withStats) throws XGBoostError {
int statsFlag = 0; int statsFlag = 0;
@ -336,7 +336,7 @@ public final class Booster {
* @param featureMap featureMap file * @param featureMap featureMap file
* @param withStats Controls whether the split statistics are output. * @param withStats Controls whether the split statistics are output.
* @return dumped model information * @return dumped model information
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public String[] getDumpInfo(String featureMap, boolean withStats) throws XGBoostError { public String[] getDumpInfo(String featureMap, boolean withStats) throws XGBoostError {
int statsFlag = 0; int statsFlag = 0;
@ -353,10 +353,10 @@ public final class Booster {
* @param modelPath file to save dumped model info * @param modelPath file to save dumped model info
* @param withStats bool * @param withStats bool
Controls whether the split statistics are output. Controls whether the split statistics are output.
* @throws FileNotFoundException * @throws FileNotFoundException file not found
* @throws UnsupportedEncodingException * @throws UnsupportedEncodingException unsupported feature
* @throws IOException * @throws IOException error with model writing
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void dumpModel(String modelPath, boolean withStats) throws FileNotFoundException, UnsupportedEncodingException, IOException, XGBoostError { public void dumpModel(String modelPath, boolean withStats) throws FileNotFoundException, UnsupportedEncodingException, IOException, XGBoostError {
File tf = new File(modelPath); File tf = new File(modelPath);
@ -380,10 +380,10 @@ public final class Booster {
* @param featureMap featureMap file * @param featureMap featureMap file
* @param withStats bool * @param withStats bool
Controls whether the split statistics are output. Controls whether the split statistics are output.
* @throws FileNotFoundException * @throws FileNotFoundException exception
* @throws UnsupportedEncodingException * @throws UnsupportedEncodingException exception
* @throws IOException * @throws IOException exception
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void dumpModel(String modelPath, String featureMap, boolean withStats) throws FileNotFoundException, UnsupportedEncodingException, IOException, XGBoostError { public void dumpModel(String modelPath, String featureMap, boolean withStats) throws FileNotFoundException, UnsupportedEncodingException, IOException, XGBoostError {
File tf = new File(modelPath); File tf = new File(modelPath);
@ -404,7 +404,7 @@ public final class Booster {
/** /**
* get importance of each feature * get importance of each feature
* @return featureMap key: feature index, value: feature importance score * @return featureMap key: feature index, value: feature importance score
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public Map<String, Integer> getFeatureScore() throws XGBoostError { public Map<String, Integer> getFeatureScore() throws XGBoostError {
String[] modelInfos = getDumpInfo(false); String[] modelInfos = getDumpInfo(false);
@ -433,7 +433,7 @@ public final class Booster {
* get importance of each feature * get importance of each feature
* @param featureMap file to save dumped model info * @param featureMap file to save dumped model info
* @return featureMap key: feature index, value: feature importance score * @return featureMap key: feature index, value: feature importance score
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public Map<String, Integer> getFeatureScore(String featureMap) throws XGBoostError { public Map<String, Integer> getFeatureScore(String featureMap) throws XGBoostError {
String[] modelInfos = getDumpInfo(featureMap, false); String[] modelInfos = getDumpInfo(featureMap, false);

36
java/xgboost4j/src/main/java/org/dmlc/xgboost4j/DMatrix.java
View File

@ -51,8 +51,8 @@ public class DMatrix {
/** /**
* init DMatrix from file (svmlight format) * init DMatrix from file (svmlight format)
* @param dataPath * @param dataPath path of data file
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public DMatrix(String dataPath) throws XGBoostError { public DMatrix(String dataPath) throws XGBoostError {
if(dataPath == null) { if(dataPath == null) {
@ -69,7 +69,7 @@ public class DMatrix {
* @param indices Indices (colIndexs for CSR or rowIndexs for CSC) * @param indices Indices (colIndexs for CSR or rowIndexs for CSC)
* @param data non zero values (sequence by row for CSR or by col for CSC) * @param data non zero values (sequence by row for CSR or by col for CSC)
* @param st sparse matrix type (CSR or CSC) * @param st sparse matrix type (CSR or CSC)
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public DMatrix(long[] headers, int[] indices, float[] data, SparseType st) throws XGBoostError { public DMatrix(long[] headers, int[] indices, float[] data, SparseType st) throws XGBoostError {
long[] out = new long[1]; long[] out = new long[1];
@ -90,7 +90,7 @@ public class DMatrix {
* @param data data values * @param data data values
* @param nrow number of rows * @param nrow number of rows
* @param ncol number of columns * @param ncol number of columns
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public DMatrix(float[] data, int nrow, int ncol) throws XGBoostError { public DMatrix(float[] data, int nrow, int ncol) throws XGBoostError {
long[] out = new long[1]; long[] out = new long[1];
@ -110,7 +110,7 @@ public class DMatrix {
/** /**
* set label of dmatrix * set label of dmatrix
* @param labels * @param labels labels
*/ */
public void setLabel(float[] labels) throws XGBoostError { public void setLabel(float[] labels) throws XGBoostError {
ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetFloatInfo(handle, "label", labels)); ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetFloatInfo(handle, "label", labels));
@ -118,8 +118,8 @@ public class DMatrix {
/** /**
* set weight of each instance * set weight of each instance
* @param weights * @param weights weights
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void setWeight(float[] weights) throws XGBoostError { public void setWeight(float[] weights) throws XGBoostError {
ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetFloatInfo(handle, "weight", weights)); ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetFloatInfo(handle, "weight", weights));
@ -128,8 +128,8 @@ public class DMatrix {
/** /**
* if specified, xgboost will start from this init margin * if specified, xgboost will start from this init margin
* can be used to specify initial prediction to boost from * can be used to specify initial prediction to boost from
* @param baseMargin * @param baseMargin base margin
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void setBaseMargin(float[] baseMargin) throws XGBoostError { public void setBaseMargin(float[] baseMargin) throws XGBoostError {
ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetFloatInfo(handle, "base_margin", baseMargin)); ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetFloatInfo(handle, "base_margin", baseMargin));
@ -138,8 +138,8 @@ public class DMatrix {
/** /**
* if specified, xgboost will start from this init margin * if specified, xgboost will start from this init margin
* can be used to specify initial prediction to boost from * can be used to specify initial prediction to boost from
* @param baseMargin * @param baseMargin base margin
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void setBaseMargin(float[][] baseMargin) throws XGBoostError { public void setBaseMargin(float[][] baseMargin) throws XGBoostError {
float[] flattenMargin = flatten(baseMargin); float[] flattenMargin = flatten(baseMargin);
@ -148,8 +148,8 @@ public class DMatrix {
/** /**
* Set group sizes of DMatrix (used for ranking) * Set group sizes of DMatrix (used for ranking)
* @param group * @param group group size as array
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public void setGroup(int[] group) throws XGBoostError { public void setGroup(int[] group) throws XGBoostError {
ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetGroup(handle, group)); ErrorHandle.checkCall(XgboostJNI.XGDMatrixSetGroup(handle, group));
@ -170,7 +170,7 @@ public class DMatrix {
/** /**
* get label values * get label values
* @return label * @return label
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[] getLabel() throws XGBoostError { public float[] getLabel() throws XGBoostError {
return getFloatInfo("label"); return getFloatInfo("label");
@ -179,7 +179,7 @@ public class DMatrix {
/** /**
* get weight of the DMatrix * get weight of the DMatrix
* @return weights * @return weights
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[] getWeight() throws XGBoostError { public float[] getWeight() throws XGBoostError {
return getFloatInfo("weight"); return getFloatInfo("weight");
@ -188,7 +188,7 @@ public class DMatrix {
/** /**
* get base margin of the DMatrix * get base margin of the DMatrix
* @return base margin * @return base margin
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public float[] getBaseMargin() throws XGBoostError { public float[] getBaseMargin() throws XGBoostError {
return getFloatInfo("base_margin"); return getFloatInfo("base_margin");
@ -198,7 +198,7 @@ public class DMatrix {
* Slice the DMatrix and return a new DMatrix that only contains `rowIndex`. * Slice the DMatrix and return a new DMatrix that only contains `rowIndex`.
* @param rowIndex * @param rowIndex
* @return sliced new DMatrix * @return sliced new DMatrix
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public DMatrix slice(int[] rowIndex) throws XGBoostError { public DMatrix slice(int[] rowIndex) throws XGBoostError {
long[] out = new long[1]; long[] out = new long[1];
@ -211,7 +211,7 @@ public class DMatrix {
/** /**
* get the row number of DMatrix * get the row number of DMatrix
* @return number of rows * @return number of rows
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public long rowNum() throws XGBoostError { public long rowNum() throws XGBoostError {
long[] rowNum = new long[1]; long[] rowNum = new long[1];

2
java/xgboost4j/src/main/java/org/dmlc/xgboost4j/util/ErrorHandle.java
View File

@ -40,7 +40,7 @@ public class ErrorHandle {
/** /**
* check the return value of C API * check the return value of C API
* @param ret return valud of xgboostJNI C API call * @param ret return valud of xgboostJNI C API call
* @throws org.dmlc.xgboost4j.util.XGBoostError * @throws org.dmlc.xgboost4j.util.XGBoostError Native error
*/ */
public static void checkCall(int ret) throws XGBoostError { public static void checkCall(int ret) throws XGBoostError {
if(ret != 0) { if(ret != 0) {

10
java/xgboost4j/src/main/java/org/dmlc/xgboost4j/util/Initializer.java
View File

@ -44,8 +44,8 @@ public class Initializer {
/** /**
* load native library, this method will first try to load library from java.library.path, then try to load library in jar package. * load native library, this method will first try to load library from java.library.path, then try to load library in jar package.
* @param libName * @param libName library path
* @throws IOException * @throws IOException exception
*/ */
private static void smartLoad(String libName) throws IOException { private static void smartLoad(String libName) throws IOException {
addNativeDir(nativePath); addNativeDir(nativePath);
@ -63,9 +63,9 @@ public class Initializer {
} }
/** /**
* add libPath to java.library.path, then native library in libPath would be load properly * Add libPath to java.library.path, then native library in libPath would be load properly
* @param libPath * @param libPath library path
* @throws IOException * @throws IOException exception
*/ */
public static void addNativeDir(String libPath) throws IOException { public static void addNativeDir(String libPath) throws IOException {
try { try {

18
java/xgboost4j/src/main/java/org/dmlc/xgboost4j/util/NativeUtils.java
View File

@ -25,10 +25,11 @@ import java.io.OutputStream;
/** /**
* Simple library class for working with JNI (Java Native Interface) * Simple library class for working with JNI (Java Native Interface)
* * <p>
* @see http://adamheinrich.com/2012/how-to-load-native-jni-library-from-jar * See <a href="http://adamheinrich.com/2012/how-to-load-native-jni-library-from-jar">
* * http://adamheinrich.com/2012/how-to-load-native-jni-library-from-jar</a>
* @author Adam Heirnich &lt;adam@adamh.cz&gt;, http://www.adamh.cz * <p>
* Author Adam Heirnich &lt;adam@adamh.cz&gt;, http://www.adamh.cz
*/ */
public class NativeUtils { public class NativeUtils {
@ -40,14 +41,17 @@ public class NativeUtils {
/** /**
* Loads library from current JAR archive * Loads library from current JAR archive
* * <p>
* The file from JAR is copied into system temporary directory and then loaded. The temporary file is deleted after exiting. * The file from JAR is copied into system temporary directory and then loaded.
* The temporary file is deleted after exiting.
* Method uses String as filename because the pathname is "abstract", not system-dependent. * Method uses String as filename because the pathname is "abstract", not system-dependent.
* <p>
* The restrictions of {@link File#createTempFile(java.lang.String, java.lang.String)} apply to {@code path}.
* *
* @param path The filename inside JAR as absolute path (beginning with '/'), e.g. /package/File.ext * @param path The filename inside JAR as absolute path (beginning with '/'), e.g. /package/File.ext
* @throws IOException If temporary file creation or read/write operation fails * @throws IOException If temporary file creation or read/write operation fails
* @throws IllegalArgumentException If source file (param path) does not exist * @throws IllegalArgumentException If source file (param path) does not exist
* @throws IllegalArgumentException If the path is not absolute or if the filename is shorter than three characters (restriction of {@see File#createTempFile(java.lang.String, java.lang.String)}). * @throws IllegalArgumentException If the path is not absolute or if the filename is shorter than three characters
*/ */
public static void loadLibraryFromJar(String path) throws IOException { public static void loadLibraryFromJar(String path) throws IOException {

3
java/xgboost4j/src/main/java/org/dmlc/xgboost4j/util/Trainer.java
View File

@ -45,6 +45,7 @@ public class Trainer {
* @param obj customized objective (set to null if not used) * @param obj customized objective (set to null if not used)
* @param eval customized evaluation (set to null if not used) * @param eval customized evaluation (set to null if not used)
* @return trained booster * @return trained booster
* @throws org.dmlc.xgboost4j.util.XGBoostError native error
*/ */
public static Booster train(Iterable<Entry<String, Object>> params, DMatrix dtrain, int round, public static Booster train(Iterable<Entry<String, Object>> params, DMatrix dtrain, int round,
Iterable<Entry<String, DMatrix>> watchs, IObjective obj, IEvaluation eval) throws XGBoostError { Iterable<Entry<String, DMatrix>> watchs, IObjective obj, IEvaluation eval) throws XGBoostError {
@ -111,6 +112,7 @@ public class Trainer {
* @param obj customized objective (set to null if not used) * @param obj customized objective (set to null if not used)
* @param eval customized evaluation (set to null if not used) * @param eval customized evaluation (set to null if not used)
* @return evaluation history * @return evaluation history
* @throws org.dmlc.xgboost4j.util.XGBoostError native error
*/ */
public static String[] crossValiation(Iterable<Entry<String, Object>> params, DMatrix data, int round, int nfold, String[] metrics, IObjective obj, IEvaluation eval) throws XGBoostError { public static String[] crossValiation(Iterable<Entry<String, Object>> params, DMatrix data, int round, int nfold, String[] metrics, IObjective obj, IEvaluation eval) throws XGBoostError {
CVPack[] cvPacks = makeNFold(data, nfold, params, metrics); CVPack[] cvPacks = makeNFold(data, nfold, params, metrics);
@ -148,6 +150,7 @@ public class Trainer {
* @param params booster parameters * @param params booster parameters
* @param evalMetrics Evaluation metrics * @param evalMetrics Evaluation metrics
* @return CV package array * @return CV package array
* @throws org.dmlc.xgboost4j.util.XGBoostError native error
*/ */
public static CVPack[] makeNFold(DMatrix data, int nfold, Iterable<Entry<String, Object>> params, String[] evalMetrics) throws XGBoostError { public static CVPack[] makeNFold(DMatrix data, int nfold, Iterable<Entry<String, Object>> params, String[] evalMetrics) throws XGBoostError {
List<Integer> samples = genRandPermutationNums(0, (int) data.rowNum()); List<Integer> samples = genRandPermutationNums(0, (int) data.rowNum());