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

ok

Browse Source
This commit is contained in:
tqchen 2014-05-25 10:15:57 -07:00
parent 1048561ede
commit 86515a2c15
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/xgboost.py
View File

@ -121,8 +121,6 @@ class Booster:
assert isinstance(d,DMatrix) assert isinstance(d,DMatrix)
dmats = ( ctypes.c_void_p * len(cache) )(*[ d.handle for d in cache]) dmats = ( ctypes.c_void_p * len(cache) )(*[ d.handle for d in cache])
self.handle = ctypes.c_void_p( xglib.XGBoosterCreate( dmats, len(cache) ) ) self.handle = ctypes.c_void_p( xglib.XGBoosterCreate( dmats, len(cache) ) )
# default random seed
self.set_param( {"seed": 0} )
self.set_param( params ) self.set_param( params )
def __del__(self): def __del__(self):
xglib.XGBoosterFree(self.handle) xglib.XGBoosterFree(self.handle)
@ -188,7 +186,7 @@ class Booster:
def train(params, dtrain, num_boost_round = 10, evals = [], obj=None): def train(params, dtrain, num_boost_round = 10, evals = [], obj=None):
""" train a booster with given paramaters """ """ train a booster with given paramaters """
bst = Booster(params, [dtrain] ) bst = Booster(params, [dtrain]+[ d[0] for d in evals ] )
if obj == None: if obj == None:
for i in range(num_boost_round): for i in range(num_boost_round):
bst.update( dtrain ) bst.update( dtrain )