s434788/ium_434788
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Firs version o ex.9

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-23 14:33:43 +02:00
parent 3955e7ac20
commit c9b0f021c1
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Jenkinsfile_train
View File

@ -31,15 +31,14 @@ pipeline {
steps steps
{ {
catchError { catchError {
sh 'python3.8 Zadanie_06_and_07_training.py ${BATCH_SIZE} ${EPOCHS}' sh 'python3.8 Zadanie_08_and_09_MLflow.py ${BATCH_SIZE} ${EPOCHS}'
} }
} }
} }
stage('Archive artifacts') { stage('Archive artifacts') {
steps{ steps{
archiveArtifacts 'wine_model.h5' archiveArtifacts 'my_model/**'
archiveArtifacts 'my_runs/**'
} }
} }
} }

2
MLproject
View File

@ -8,4 +8,4 @@ entry_points:
parameters: parameters:
batch_size: {type: int, default: 16} batch_size: {type: int, default: 16}
epochs: {type: int, default: 15} epochs: {type: int, default: 15}
command: "python3 Zadanie_08_MLflow.py {batch_size} {epochs}" command: "python3 Zadanie_08_and_09_MLflow.py {batch_size} {epochs}"

5
Zadanie_08_MLflow.py → Zadanie_08_and_09_MLflow.py
View File

@ -17,6 +17,7 @@ with mlflow.start_run():
batch_param = int(sys.argv[1]) if len(sys.argv) > 1 else 16 batch_param = int(sys.argv[1]) if len(sys.argv) > 1 else 16
epoch_param = int(sys.argv[2]) if len(sys.argv) > 2 else 15 epoch_param = int(sys.argv[2]) if len(sys.argv) > 2 else 15
mlflow.set_experiment("s434788")
mlflow.log_param("batch_size", batch_param) mlflow.log_param("batch_size", batch_param)
mlflow.log_param("epochs", epoch_param) mlflow.log_param("epochs", epoch_param)
@ -69,4 +70,6 @@ with mlflow.start_run():
print(f"accuracy: {results}") print(f"accuracy: {results}")
mlflow.log_metric("Accuracy", results) mlflow.log_metric("Accuracy", results)
mlflow.keras.save_model(NeuralModel, "my_model")