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

change initTracker() to static

Browse Source
This commit is contained in:
CodingCat 2016-03-04 10:55:02 -05:00
parent 10a1517502
commit 416e1434e7
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/RabitTracker.java
View File

@ -24,6 +24,15 @@ public class RabitTracker {
private int num_workers;
private AtomicReference<Process> trackerProcess = new AtomicReference<Process>();
static {
try {
initTrackerPy();
} catch (IOException ex) {
logger.error("load tracker library failed.");
logger.error(ex);
}
}
/**
* Tracker logger that logs output from tracker.
*/
@ -44,7 +53,7 @@ public class RabitTracker {
}
}
private void initTrackerPy() throws IOException {
private static void initTrackerPy() throws IOException {
try {
tracker_py = NativeLibLoader.createTempFileFromResource("/tracker.py");
} catch (IOException ioe) {
@ -90,7 +99,6 @@ public class RabitTracker {
private boolean startTrackerProcess() {
try {
initTrackerPy();
trackerProcess.set(Runtime.getRuntime().exec("python " + tracker_py +
" --num-workers=" + String.valueOf(num_workers)));
loadEnvs(trackerProcess.get().getInputStream());