mlflow fix5
Some checks failed
s444452-training/pipeline/head There was a failure building this commit
Some checks failed
s444452-training/pipeline/head There was a failure building this commit
This commit is contained in:
parent
ad29768ab6
commit
d4e1066199
@ -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
|
|
@ -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) {
|
||||||
|
Loading…
Reference in New Issue
Block a user