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

add logs
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 15:02:40 +02:00
parent bc8d6c5b70
commit eef645fc49
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lab8/trainScript.py
View File

@ -100,13 +100,13 @@ def train():
# pred = np.array(linear_model.predict(feature_test_sample))
# flatten_pred = flatten(pred)
#
## with open(cwd + "/../result.txt", "w+") as resultFile:
# with open(cwd + "/../result.txt", "w+") as resultFile:
# resultFile.write("predictions: " + str(flatten_pred) + '\n')
# resultFile.write("expected: " + str(labels_test_sample.to_numpy()))
# mlflow.log_param('epochs', numberOfEpochParam)
# mlflow.log_param('learning_rate', learning_rate)
# mlflow.log_metric('final_loss', min(hist["val_loss"]))
mlflow.log_param('epochs', numberOfEpochParam)
mlflow.log_param('learning_rate', learning_rate)
mlflow.log_metric('final_loss', min(hist["val_loss"]))
signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))