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

[jvm-packages] delete all constraints from spark layer about obj and eval metrics and handle error in jvm layer (#4560)

Browse Source 
* temp

* prediction part

* remove supported*

* add for test

* fix param name

* add rabit

* update rabit

* return value of rabit init

* eliminate compilation warnings

* update rabit

* shutdown

* update rabit again

* check sparkcontext shutdown

* fix logic

* sleep

* fix tests

* test with relaxed threshold

* create new thread each time

* stop for job quitting

* udpate rabit

* update rabit

* update rabit

* update git modules
This commit is contained in:
Nan Zhu
2019-06-27 08:47:37 -07:00
committed by GitHub
parent dd01f7c4f5
commit abffbe014e
11 changed files with 143 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -818,8 +818,11 @@ JNIEXPORT jint JNICALL Java_ml_dmlc_xgboost4j_java_XGBoostJNI_RabitInit
argv.push_back(&args[i][0]);
}
RabitInit(args.size(), dmlc::BeginPtr(argv));
return 0;
if (RabitInit(args.size(), dmlc::BeginPtr(argv))) {
return 0;
} else {
return 1;
}
}
/*
@@ -829,8 +832,11 @@ JNIEXPORT jint JNICALL Java_ml_dmlc_xgboost4j_java_XGBoostJNI_RabitInit
*/
JNIEXPORT jint JNICALL Java_ml_dmlc_xgboost4j_java_XGBoostJNI_RabitFinalize
(JNIEnv *jenv, jclass jcls) {
RabitFinalize();
return 0;
if (RabitFinalize()) {
return 0;
} else {
return 1;
}
}
/*