s434695/ium_434695
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

change for Domino2
All checks were successful
s434695-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
s434695 2021-05-23 19:56:42 +02:00
parent caffb7a796
commit f8addf09d4
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Jenkinsfile vendored
View File

@ -33,7 +33,7 @@ pipeline {
{ {
sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}' sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
sh 'rm -r my_model' sh 'rm -r my_model'
sh 'python3 vgsales-mlflow.py ${BATCH_SIZE} ${EPOCHS}' sh 'python3 vgsales-mlflow.py'
} }
} }

1
vgsales-mlflow.py
View File

@ -66,4 +66,3 @@ with mlflow.start_run():
mlflow.log_metric("rmse", rmse) mlflow.log_metric("rmse", rmse)
#mlflow.keras.log_model(model, 'vgsales_model.h5') #mlflow.keras.log_model(model, 'vgsales_model.h5')
mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train) mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train)