s426206/ium_426206
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Registred model.

Browse Source
This commit is contained in:
Jan Nowak 2021-05-23 15:36:42 +02:00
parent fe85610850
commit 63cb9a7d84
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
generate_MLmodel.py
View File

@ -29,6 +29,7 @@ if __name__ == "__main__":
lr = args.lr
n_epochs = args.epochs
mlflow.set_tracking_uri("http://127.0.0.1:5000")
mlflow.set_experiment("s426206")
with mlflow.start_run():
mlflow.log_param("lr", lr)
@ -129,7 +130,7 @@ if __name__ == "__main__":
siganture = infer_signature(x_train, model(torch.tensor(np.reshape(x_train, (-1,1))).float()).numpy())
#mlflow.set_experiment("s426206")
mlflow.set_tracking_uri("http://172.17.0.1:5000")
#mlflow.set_tracking_uri("http://172.17.0.1:5000")
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
# print(tracking_url_type_store)
# Model registry does not work with file store