Skip to content

Commit

Permalink
Update xgboost.py
Browse files Browse the repository at this point in the history
  • Loading branch information
tqchen committed May 6, 2015
1 parent 60bf389 commit 2d748fb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions wrapper/xgboost.py
Original file line number Diff line number Diff line change
Expand Up @@ -478,7 +478,7 @@ def save_model(self, fname):
Parameters
----------
fname : string
Output file name or handle
Output file name
"""
if isinstance(fname, string_types): # assume file name
xglib.XGBoosterSaveModel(self.handle, c_str(fname))
Expand All @@ -504,7 +504,7 @@ def load_model(self, fname):
Parameters
----------
fname : string of file handle
fname : string or a memory buffer
Input file name or memory buffer(see also save_raw)
"""
if isinstance(fname, str): # assume file name
Expand Down

0 comments on commit 2d748fb

Please sign in to comment.