add one more layer to model dir

2023-04-11 11:56:56 +08:00
parent 0e6a5b8925
commit ac6400e93a
2 changed files with 2 additions and 2 deletions
--- a/log_unsupported_model.py
+++ b/log_unsupported_model.py
@ -64,7 +64,7 @@ def pull_model(server_uri: str,
    print(f'input data: {fake_data}, predictions: {output}')

    # save it to local file system
-    mlflow.pyfunc.save_model(path='./model', python_model=model)
+    mlflow.pyfunc.save_model(path=f'./model/{exp_name}', python_model=model)


 if __name__ == '__main__':