s444417/ium_444417
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

comment loss
All checks were successful
s444417-training/pipeline/head This commit looks good
Details
s444417-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
s444417 2022-05-14 14:55:29 +02:00
parent c3e3923197
commit bf499f6f09
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lab8/trainScript.py
View File

@ -106,7 +106,7 @@ def train():
mlflow.log_param('epochs', numberOfEpochParam) mlflow.log_param('epochs', numberOfEpochParam)
mlflow.log_param('learning_rate', learning_rate) mlflow.log_param('learning_rate', learning_rate)
mlflow.log_metric('final_loss', min(hist["val_loss"])) # mlflow.log_metric('final_loss', min(hist["val_loss"]))
signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features)) signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))