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

fix trainScript line tabs
Some checks failed
s444417-training/pipeline/head This commit looks good
Details
s444417-evaluation/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
s444417 2022-05-20 18:16:52 +02:00
parent f4457b83c6
commit ce9ca84ddf
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
lab8/trainScript.py
View File

@ -108,16 +108,16 @@ def train():
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))
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
sampleInp = [0.0, 0.0, 2.0, 904.129525, 1.000000, 1.000000, 20.098413, 79.107860]
# expected value is 49.7
if tracking_url_type_store != "file":
mlflow.keras.log_model(linear_model, "linear-model", registered_model_name="HousePriceLinear", signature=signature)
else:
mlflow.keras.log_model(linear_model, "model", signature=signature, input_example=np.array(sampleInp))
signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
sampleInp = [0.0, 0.0, 2.0, 904.129525, 1.000000, 1.000000, 20.098413, 79.107860]
# expected value is 49.7
if tracking_url_type_store != "file":
mlflow.keras.log_model(linear_model, "linear-model", registered_model_name="HousePriceLinear", signature=signature)
else:
mlflow.keras.log_model(linear_model, "model", signature=signature, input_example=np.array(sampleInp))
if __name__ == '__main__':
train()