From 52c4f9a6820056ec23ca9fe0650680d2c0121a11 Mon Sep 17 00:00:00 2001 From: tqchen Date: Sat, 17 May 2014 09:13:54 -0700 Subject: [PATCH] add return type for xgboost, don't know if it is mac problem --- python/xgboost.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/xgboost.py b/python/xgboost.py index d7cf9f63e21d..3be5a2b066e5 100644 --- 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 ):