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

allow common python output in single node

Browse Source
This commit is contained in:
tqchen 2016-04-11 15:47:45 -07:00
parent 59610c49df
commit 49f3892942
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
python-package/xgboost/rabit.py
View File

@ -12,6 +12,7 @@ def _init_rabit():
if _LIB is not None: if _LIB is not None:
_LIB.RabitGetRank.restype = ctypes.c_int _LIB.RabitGetRank.restype = ctypes.c_int
_LIB.RabitGetWorldSize.restype = ctypes.c_int _LIB.RabitGetWorldSize.restype = ctypes.c_int
_LIB.RabitIsDistributed.restype = ctypes.c_int
_LIB.RabitVersionNumber.restype = ctypes.c_int _LIB.RabitVersionNumber.restype = ctypes.c_int
def init(args=None): def init(args=None):
@ -65,8 +66,12 @@ def tracker_print(msg):
""" """
if not isinstance(msg, STRING_TYPES): if not isinstance(msg, STRING_TYPES):
msg = str(msg) msg = str(msg)
_LIB.RabitTrackerPrint(c_str(msg)) is_dist = _LIB.RabitIsDistributed()
if is_dist != 0:
_LIB.RabitTrackerPrint(c_str(msg))
else:
sys.stdout.write(msg)
sys.stdout.flush()
def get_processor_name(): def get_processor_name():
"""Get the processor name. """Get the processor name.

2
rabit

@ -1 +1 @@
Subproject commit be50e7b63224b9fb7ff94ce34df9f8752ef83043 Subproject commit 849b20b7c822d194a515cc5587c37764cdf39385