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

[jvm-packages] Add getNumFeature method (#6075)

Browse Source 
* Add getNumFeature to the Java API
* Add getNumFeature to the Scala API
* Add unit tests for getNumFeature

Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Hristo Iliev
2020-09-08 06:57:46 +03:00
committed by GitHub
parent 93e9af43bb
commit da61d9460b
7 changed files with 66 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
jvm-packages/xgboost4j/src/native/xgboost4j.cpp
View File

@@ -848,6 +848,22 @@ JNIEXPORT jint JNICALL Java_ml_dmlc_xgboost4j_java_XGBoostJNI_XGBoosterSaveRabit
return XGBoosterSaveRabitCheckpoint(handle);
}
/*
* Class: ml_dmlc_xgboost4j_java_XGBoostJNI
* Method: XGBoosterGetNumFeature
* Signature: (J[J)I
*/
JNIEXPORT jint JNICALL Java_ml_dmlc_xgboost4j_java_XGBoostJNI_XGBoosterGetNumFeature
(JNIEnv *jenv, jclass jcls, jlong jhandle, jlongArray jout) {
BoosterHandle handle = (BoosterHandle) jhandle;
bst_ulong num_feature;
int ret = XGBoosterGetNumFeature(handle, &num_feature);
JVM_CHECK_CALL(ret);
jlong jnum_feature = num_feature;
jenv->SetLongArrayRegion(jout, 0, 1, &jnum_feature);
return ret;
}
/*
* Class: ml_dmlc_xgboost4j_java_XGBoostJNI
* Method: RabitInit