Update xgboost.py

This commit is contained in:
Tianqi Chen 2014-08-27 12:27:25 -07:00
parent 2451ba0f1c
commit 57c0ab2721

View File

@ -12,7 +12,7 @@ import scipy.sparse as scp
if os.name == 'nt': if os.name == 'nt':
XGBOOST_PATH = os.path.dirname(__file__)+'/../windows/x64/Release/xgboost_wrapper.dll' XGBOOST_PATH = os.path.dirname(__file__)+'/../windows/x64/Release/xgboost_wrapper.dll'
else: else:
XGBOOST_PATH = os.path.dirname(__file__)+'/../libxgboostwrapper.so' XGBOOST_PATH = os.path.dirname(__file__)+'/../libxgboostwrapper.so'
# load in xgboost library # load in xgboost library
xglib = ctypes.cdll.LoadLibrary(XGBOOST_PATH) xglib = ctypes.cdll.LoadLibrary(XGBOOST_PATH)