mlflow fix5
Some checks failed
s444452-training/pipeline/head There was a failure building this commit

This commit is contained in:
AdamOsiowy123 2022-05-15 23:21:05 +02:00
parent ad29768ab6
commit d4e1066199
2 changed files with 4 additions and 2 deletions

View File

@ -10,4 +10,3 @@ COPY . /code/
RUN python3 -m pip install -r requirements.txt RUN python3 -m pip install -r requirements.txt
RUN mkdir /.kaggle && chmod o+w /.kaggle RUN mkdir /.kaggle && chmod o+w /.kaggle
RUN mkdir /code/mlruns && chmod o+w /code/mlruns

View File

@ -41,6 +41,7 @@ node {
stage('Run script') { stage('Run script') {
withEnv(["DATA_PATH=${params.DATA_PATH}","EPOCHS=${params.EPOCHS}","NUM_WORDS=${params.NUM_WORDS}", withEnv(["DATA_PATH=${params.DATA_PATH}","EPOCHS=${params.EPOCHS}","NUM_WORDS=${params.NUM_WORDS}",
"BATCH_SIZE=${params.BATCH_SIZE}","PAD_LENGTH=${params.PAD_LENGTH}"]) { "BATCH_SIZE=${params.BATCH_SIZE}","PAD_LENGTH=${params.PAD_LENGTH}"]) {
sh "chmod +x Scripts/train_neural_network.py"
sh "python3 Scripts/train_neural_network.py $DATA_PATH $EPOCHS $NUM_WORDS $BATCH_SIZE $PAD_LENGTH" sh "python3 Scripts/train_neural_network.py $DATA_PATH $EPOCHS $NUM_WORDS $BATCH_SIZE $PAD_LENGTH"
} }
} }
@ -48,6 +49,8 @@ node {
archiveArtifacts "model/neural_net" archiveArtifacts "model/neural_net"
archiveArtifacts "my_runs/**" archiveArtifacts "my_runs/**"
archiveArtifacts "mlruns/**" archiveArtifacts "mlruns/**"
archiveArtifacts "/mlruns/**"
archiveArtifacts "/code/mlruns/**"
} }
} }
} catch (e) { } catch (e) {