:bugs: fix: fix saving when using single filename with gboost predictors

This commit is contained in:
Ali Zaidi 2021-01-15 12:32:46 -08:00
Родитель a7d37b7008
Коммит a3bb51be10
1 изменённых файлов: 2 добавлений и 0 удалений

Просмотреть файл

@ -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)