updated model path reading
This commit is contained in:
Родитель
80b22456be
Коммит
4df99190eb
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче