s444501/ium_444501
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

mlflow
All checks were successful
s444501-training/pipeline/head This commit looks good
Details
s444501-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
s444501 2022-05-13 02:11:38 +02:00
parent c685f36b54
commit f8025ad3fc
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
biblioteki_ml.py
View File

@ -11,8 +11,7 @@ import pandas as pd
# MLFlow
mlflow.set_tracking_uri("http://172.17.0.1:5000")
restore_experiment('15')
mlflow.set_experiment("s444501")
mlflow.set_experiment("444501")
# Parametry z konsoli
@ -118,13 +117,13 @@ def train_main(epochs, run):
if tracking_url_type_store != "file":
mlflow.pytorch.log_model(model,
's444501',
registered_model_name='s444501',
'444501',
registered_model_name='444501',
signature=signature,
input_example=X_test.numpy())
else:
mlflow.pytorch.log_model(model,
's444501',
'444501',
signature=signature,
input_example=X_test.numpy())