s444452/ium_444452
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Jenkins/Jenkinsfile.training
View File

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

3
Scripts/train_neural_network.py
View File

@ -29,7 +29,8 @@ def tokenize(x, x_train):
def save_model(model): 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) model.save(os.path.join(os.getcwd(), 'model', model_name), save_format='h5', overwrite=True)