Compare commits

...

2 Commits

Author SHA1 Message Date
deng ac6400e93a add one more layer to model dir 2023-04-11 11:56:56 +08:00
deng 0e6a5b8925 update service 2023-04-11 11:56:07 +08:00
4 changed files with 2 additions and 2 deletions

View File

@ -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__':

View File

@ -21,7 +21,7 @@ def main():
print(my_fortune)
# save model and env to local file system
mlflow.pytorch.save_model(model, './model')
mlflow.pytorch.save_model(model, './model/fortune_predict_model')
if __name__ == '__main__':

Binary file not shown.

Binary file not shown.