This commit is contained in:
Mateusz 2024-05-04 15:59:35 +02:00
parent 9d6ffe8205
commit b1a03b41b0
2 changed files with 8 additions and 2 deletions

9
Jenkinsfile vendored
View File

@ -13,6 +13,7 @@ pipeline {
description: 'Which build to use for copying artifacts',
name: 'BUILD_SELECTOR'
)
gitParameter branchFilter: 'origin/(.*)', defaultValue: 'training', name: 'BRANCH', type: 'PT_BRANCH'
}
stages {
@ -30,7 +31,13 @@ pipeline {
stage('Copy Artifacts from training job') {
steps {
copyArtifacts filter: 'model/*', projectName: 's464913-training/training', selector: buildParameter('BUILD_SELECTOR')
copyArtifacts filter: 'model/*', projectName: 's464913-training/' + params.BRANCH, selector: buildParameter('BUILD_SELECTOR')
}
}
stage('Copy Artifacts from training job') {
steps {
copyArtifacts filter: 'evaluation/*', projectName: 's464913-evaluation/evaluation', selector: buildParameter('BUILD_SELECTOR'), optional: true
}
}

View File

@ -11,7 +11,6 @@ import numpy as np
def main():
model = load_model("model/model.keras")
X_test = pd.read_csv("data/X_test.csv")
y_test = pd.read_csv("data/y_test.csv")
y_pred = model.predict(X_test)
y_pred = y_pred >= 0.5