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

comment all params
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:58:46 +02:00
parent bf499f6f09
commit bc8d6c5b70
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lab8/trainScript.py
View File

@ -104,8 +104,8 @@ def train():
# resultFile.write("predictions: " + str(flatten_pred) + '\n') # resultFile.write("predictions: " + str(flatten_pred) + '\n')
# resultFile.write("expected: " + str(labels_test_sample.to_numpy())) # resultFile.write("expected: " + str(labels_test_sample.to_numpy()))
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))