diff --git a/Jenkins/Jenkinsfile.training b/Jenkins/Jenkinsfile.training index 4fd1426..a9ed82e 100644 --- a/Jenkins/Jenkinsfile.training +++ b/Jenkins/Jenkinsfile.training @@ -24,7 +24,7 @@ node { } } stage('Archive artifacts') { - archiveArtifacts "model/**" + archiveArtifacts "model/neural_net" } } } catch (e) { diff --git a/Scripts/train_neural_network.py b/Scripts/train_neural_network.py index a3d779e..cf35ed3 100644 --- a/Scripts/train_neural_network.py +++ b/Scripts/train_neural_network.py @@ -29,7 +29,8 @@ def tokenize(x, x_train): def save_model(model): - model_name = 'neural_net_' + datetime.datetime.today().strftime('%d-%b-%Y-%H:%M:%S') + # model_name = 'neural_net_' + datetime.datetime.today().strftime('%d-%b-%Y-%H:%M:%S') + model_name = 'neural_net' model.save(os.path.join(os.getcwd(), 'model', model_name), save_format='h5', overwrite=True)