Ejemplo n.º 1
0
 /**
  * get the dump of the model as a string array
  *
  * @param withStats Controls whether the split statistics are output.
  * @return dumped model information
  * @throws XGBoostError native error
  */
 private String[] getDumpInfo(boolean withStats) throws XGBoostError {
   int statsFlag = 0;
   if (withStats) {
     statsFlag = 1;
   }
   String[][] modelInfos = new String[1][];
   JNIErrorHandle.checkCall(XgboostJNI.XGBoosterDumpModel(handle, "", statsFlag, modelInfos));
   return modelInfos[0];
 }