fix train
All checks were successful
s444452-training/pipeline/head This commit looks good

This commit is contained in:
AdamOsiowy123 2022-05-04 12:27:39 +02:00
parent 8fa7f68080
commit bc50357b19
2 changed files with 3 additions and 2 deletions

View File

@ -24,7 +24,7 @@ node {
}
}
stage('Archive artifacts') {
archiveArtifacts "model/**"
archiveArtifacts "model/neural_net"
}
}
} catch (e) {

View File

@ -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)