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

bugfix setup

Browse Source
This commit is contained in:
Tianqi Chen 2015-04-26 00:17:58 -07:00
parent 2eb30e732d
commit 94fac1076a
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wrapper/setup.py
View File

@ -12,9 +12,10 @@ dll_path = [curr_dir]
if os.name == 'nt':
if platform.architecture()[0] == '64bit':
dll_path.append(os.path.join(curr_dir, '../windows/Release/'))
else:
dll_path.append(os.path.join(curr_dir, '../windows/x64/Release/'))
else:
dll_path.append(os.path.join(curr_dir, '../windows/Release/'))
if os.name == 'nt':
dll_path = [os.path.join(p, 'xgboost_wrapper.dll') for p in dll_path]

5
wrapper/xgboost.py
View File

@ -43,10 +43,9 @@ def load_xglib():
dll_path = [curr_path]
if os.name == 'nt':
if platform.architecture()[0] == '64bit':
dll_path.append(os.path.join(curr_dir, '../windows/Release/'))
dll_path.append(os.path.join(curr_path, '../windows/x64/Release/'))
else:
dll_path.append(os.path.join(curr_dir, '../windows/x64/Release/'))
dll_path.append(os.path.join(curr_path, '../windows/Release/'))
if os.name == 'nt':
dll_path = [os.path.join(p, 'xgboost_wrapper.dll') for p in dll_path]
else: