fixed model path
This commit is contained in:
Родитель
8e99107a0d
Коммит
a6bff2d712
|
@ -22,7 +22,7 @@ def init():
|
|||
|
||||
model_path = Model.get_model_path(args.model_name)
|
||||
print(f"Model path: {model_path}")
|
||||
model = joblib.load(model_path)
|
||||
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")
|
||||
|
@ -56,4 +56,4 @@ def run(file_list):
|
|||
result = pd.concat([sno, proba], axis=1)
|
||||
results = results.append(result)
|
||||
print(f"Batch scored: {filename}")
|
||||
return results
|
||||
return results
|
||||
|
|
Загрузка…
Ссылка в новой задаче