Merge pull request #5 from tqchen/dev

add return type for xgboost, don't know if it is mac problem. #4
dmlc · May 17, 2014 · 2c14b95 · 2c14b95
2 parents 528cc27 + 52c4f9a
commit 2c14b95
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/xgboost.py b/python/xgboost.py
@@ -22,6 +22,7 @@ class REntry(ctypes.Structure):
 xglib.XGDMatrixGetLabel.restype =  ctypes.POINTER( ctypes.c_float )
 xglib.XGDMatrixGetWeight.restype =  ctypes.POINTER( ctypes.c_float )
 xglib.XGDMatrixGetRow.restype = ctypes.POINTER( REntry )
+xglib.XGBoosterCreate.restype = ctypes.c_void_p
 xglib.XGBoosterPredict.restype = ctypes.POINTER( ctypes.c_float ) 
 
 def ctypes2numpy( cptr, length ):