BUG: incorrect model_file results in segfault

2015-09-16 21:53:51 +09:00
parent ae43fd7c7a
commit db0c9e1c2d
2 changed files with 15 additions and 1 deletions
--- a/python-package/xgboost/core.py
+++ b/python-package/xgboost/core.py
@@ -743,7 +743,10 @@ class Booster(object):
            Input file name or memory buffer(see also save_raw)
        """
        if isinstance(fname, str):  # assume file name
-            _LIB.XGBoosterLoadModel(self.handle, c_str(fname))
+            if os.path.exists(fname):
+                _LIB.XGBoosterLoadModel(self.handle, c_str(fname))
+            else:
+                raise ValueError("No such file: {0}")
        else:
            buf = fname
            length = ctypes.c_ulong(len(buf))