:bugs: fix: fix saving when using single filename with gboost predictors
This commit is contained in:
Родитель
a7d37b7008
Коммит
a3bb51be10
2
base.py
2
base.py
|
@ -180,6 +180,8 @@ class BaseModel(abc.ABC):
|
|||
|
||||
def save_model(self, filename):
|
||||
|
||||
if not any([s in filename for s in [".pkl", ".pickle"]]):
|
||||
filename += ".pkl"
|
||||
parent_dir = pathlib.Path(filename).parent
|
||||
if not parent_dir.exists():
|
||||
parent_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
|
Загрузка…
Ссылка в новой задаче