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 int num_workers;
private AtomicReference<Process> trackerProcess = new AtomicReference<Process>(); 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. * 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 { try {
tracker_py = NativeLibLoader.createTempFileFromResource("/tracker.py"); tracker_py = NativeLibLoader.createTempFileFromResource("/tracker.py");
} catch (IOException ioe) { } catch (IOException ioe) {
@ -90,7 +99,6 @@ public class RabitTracker {
private boolean startTrackerProcess() { private boolean startTrackerProcess() {
try { try {
initTrackerPy();
trackerProcess.set(Runtime.getRuntime().exec("python " + tracker_py + trackerProcess.set(Runtime.getRuntime().exec("python " + tracker_py +
" --num-workers=" + String.valueOf(num_workers))); " --num-workers=" + String.valueOf(num_workers)));
loadEnvs(trackerProcess.get().getInputStream()); loadEnvs(trackerProcess.get().getInputStream());