From e819ccf0841399ba60bc473561d24631486a1317 Mon Sep 17 00:00:00 2001 From: jakubknczny Date: Fri, 21 May 2021 16:26:48 +0200 Subject: [PATCH] mlflow save model --- lab5/train/Jenkinsfile | 1 + lab5/train/train.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/lab5/train/Jenkinsfile b/lab5/train/Jenkinsfile index f38374c..90cdb6f 100644 --- a/lab5/train/Jenkinsfile +++ b/lab5/train/Jenkinsfile @@ -46,6 +46,7 @@ pipeline { steps { archiveArtifacts 'grid-stability-dense.h5' archiveArtifacts 'my_runs/**' + archiveArtifacts 'grid-stability-dense/**' } } } diff --git a/lab5/train/train.py b/lab5/train/train.py index 041ef15..bfb4262 100644 --- a/lab5/train/train.py +++ b/lab5/train/train.py @@ -1,5 +1,6 @@ from datetime import datetime import mlflow +from mlflow.models.signature import infer_signature import pandas as pd from sacred import Experiment from sacred.observers import FileStorageObserver, MongoObserver @@ -47,6 +48,10 @@ def prepare_train_model(learning_rate, _run): history = model.fit(X_train, Y_train, epochs=2, validation_data=(X_valid, Y_valid)) model.save('grid-stability-dense.h5') + mlflow.keras.save_model(keras_model=model, + path='grid-stability-dense', + signature=infer_signature(X_train, Y_train), + input_example=X_train.iloc[0]) _run.info['history'] = str(history.history['loss'][-1]) mlflow.log_metric('loss', history.history['loss'][-1])