This commit is contained in:
Cindy Weng 2022-09-01 14:33:50 +01:00 коммит произвёл GitHub
Родитель 80b22456be
Коммит 4df99190eb
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 5 добавлений и 2 удалений

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

@ -27,7 +27,10 @@ def init():
model_path = Model.get_model_path(args.model_name)
print(f"Model path: {model_path}")
model = joblib.load(os.path.join(model_path, 'model.pkl'))
if 'model.pkl' in model_path:
model = joblib.load(model_path)
else:
model = joblib.load(os.path.join(model_path, 'model.pkl'))
# load the explainer
explainer_path = os.path.join(Model.get_model_path(args.model_name), "explainer")
@ -71,4 +74,4 @@ def run(file_list):
full_results = pd.concat(all_results)
collector.batch_collect(full_results)
return results
return results