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

archive model
Some checks failed
s444452-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
AdamOsiowy123 2022-05-02 19:03:06 +02:00
parent 9652f51020
commit 02a091883a
2 changed files with 2 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') {
archiveArtifacts "neural_network_evaluation.txt, neural_network_model/**/*"
archiveArtifacts "neural_network_evaluation.txt, neural_network_model"
}
}
}

2
Scripts/train_neural_network.py
View File

@ -41,7 +41,7 @@ def load_trained_model(abs_path, model_name):
def save_model(model, abs_path, model_name):
model.save(os.path.join(abs_path, model_name))
model.save(os.path.join(abs_path, model_name), save_format='h5')
def train_model(model, x_train, y_train):