mlflow attempt
This commit is contained in:
parent
e4fc560238
commit
0879e7ed2a
|
@ -31,6 +31,7 @@ pipeline {
|
||||||
sh 'python3 ./nn_train_mlflow.py'
|
sh 'python3 ./nn_train_mlflow.py'
|
||||||
archiveArtifacts artifacts: 'mlruns/**'
|
archiveArtifacts artifacts: 'mlruns/**'
|
||||||
sh 'rm -r mlruns'
|
sh 'rm -r mlruns'
|
||||||
|
sh 'rm -r my_model_mlflow'
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -83,6 +83,6 @@ with mlflow.start_run():
|
||||||
mlflow.log_param("1st_activation_funct", first_activation_funct)
|
mlflow.log_param("1st_activation_funct", first_activation_funct)
|
||||||
mlflow.log_param("2nd_activation_funct", second_activation_funct)
|
mlflow.log_param("2nd_activation_funct", second_activation_funct)
|
||||||
#mlflow.keras.log_model(model, 'my_model')
|
#mlflow.keras.log_model(model, 'my_model')
|
||||||
mlflow.keras.save_model(model, "my_model")
|
mlflow.keras.save_model(model, "my_model_mlflow")
|
||||||
mlflow.log_metric("accuracy", accuracy_score(y_true, y_pred))
|
mlflow.log_metric("accuracy", accuracy_score(y_true, y_pred))
|
||||||
|
|
Loading…
Reference in New Issue