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

CLN: Remove unused import. Fix comment.

Browse Source
This commit is contained in:
Skipper Seabold 2015-05-08 09:25:55 -05:00
parent a4de0ebcd4
commit 932af821c5
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
wrapper/xgboost.py
View File

@ -15,7 +15,6 @@ import re
import ctypes import ctypes
import platform import platform
import collections import collections
from io import BytesIO
import numpy as np import numpy as np
import scipy.sparse import scipy.sparse
@ -492,7 +491,7 @@ class Booster(object):
def save_raw(self): def save_raw(self):
""" """
Save the model to a in memory buffer represetation Save the model to a in memory buffer represetation
Returns Returns
------- -------
a in memory buffer represetation of the model a in memory buffer represetation of the model
@ -876,12 +875,12 @@ class XGBModel(XGBModelBase):
self._Booster = None self._Booster = None
def __getstate__(self): def __getstate__(self):
# can't pickle ctypes pointers so put _Booster in a BytesIO obj # can't pickle ctypes pointers so put _Booster in a bytearray object
this = self.__dict__.copy() # don't modify in place this = self.__dict__.copy() # don't modify in place
bst = this["_Booster"] bst = this["_Booster"]
if bst is not None: if bst is not None:
raw = this["_Booster"].save_raw() raw = this["_Booster"].save_raw()
this["_Booster"] = raw this["_Booster"] = raw
return this return this
def __setstate__(self, state): def __setstate__(self, state):
@ -894,7 +893,7 @@ class XGBModel(XGBModelBase):
""" """
get the underlying xgboost Booster of this model get the underlying xgboost Booster of this model
will raise an exception when fit was not called will raise an exception when fit was not called
Returns Returns
------- -------
booster : a xgboost booster of underlying model booster : a xgboost booster of underlying model
@ -902,7 +901,7 @@ class XGBModel(XGBModelBase):
if self._Booster is None: if self._Booster is None:
raise XGBError('need to call fit beforehand') raise XGBError('need to call fit beforehand')
return self._Booster return self._Booster
def get_xgb_params(self): def get_xgb_params(self):
xgb_params = self.get_params() xgb_params = self.get_params()