Fix 3
Some checks failed
s434804-training/pipeline/head There was a failure building this commit

This commit is contained in:
Dawid 2021-05-23 18:10:53 +02:00
parent d8d81df090
commit f6465a49f6
2 changed files with 3 additions and 1 deletions

View File

@ -27,7 +27,7 @@ pipeline {
stage('archiveArtifacts') { stage('archiveArtifacts') {
steps{ steps{
archiveArtifacts 'vaccines_model/saved_model.pb' archiveArtifacts 'vaccines_model/saved_model.pb'
archiveArtifacts 'country_vaccination/**' archiveArtifacts 'country_vaccination2/**'
} }
} }
stage('sendMail') { stage('sendMail') {

View File

@ -57,8 +57,10 @@ if __name__ == "__main__":
mlf.set_experiment("s434804") mlf.set_experiment("s434804")
tracking_url_type_store = urlparse(mlf.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlf.get_tracking_uri()).scheme
if tracking_url_type_store != "file": if tracking_url_type_store != "file":
print("no file")
mlf.keras.log_model(model, "country_vaccination2", registered_model_name="s434804", signature=signature, mlf.keras.log_model(model, "country_vaccination2", registered_model_name="s434804", signature=signature,
input_example=input_example) input_example=input_example)
else: else:
print("file")
mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example) mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example)
mlf.keras.save_model(model, "country_vaccination", signature=signature, input_example=input_example) mlf.keras.save_model(model, "country_vaccination", signature=signature, input_example=input_example)