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

Update xgboost.py

Browse Source
This commit is contained in:
Tianqi Chen 2015-05-06 16:46:27 -07:00
parent 60bf389825
commit 2d748fb6fa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wrapper/xgboost.py
View File

@ -478,7 +478,7 @@ class Booster(object):
Parameters Parameters
---------- ----------
fname : string fname : string
Output file name or handle Output file name
""" """
if isinstance(fname, string_types): # assume file name if isinstance(fname, string_types): # assume file name
xglib.XGBoosterSaveModel(self.handle, c_str(fname)) xglib.XGBoosterSaveModel(self.handle, c_str(fname))
@ -504,7 +504,7 @@ class Booster(object):
Parameters Parameters
---------- ----------
fname : string of file handle fname : string or a memory buffer
Input file name or memory buffer(see also save_raw) Input file name or memory buffer(see also save_raw)
""" """
if isinstance(fname, str): # assume file name if isinstance(fname, str): # assume file name