s444018/ium_444018
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix dllib.py call
Some checks failed
s444018-evaluation/pipeline/head This commit looks good
Details
s444018-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Szymon Parafiński 2022-05-16 02:40:30 +02:00
parent bcf2dd1c65
commit 5a7e92fd4d
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Jenkinsfile_train
View File

@ -21,8 +21,8 @@ pipeline {
}
stage('Train model with sacred') {
steps {
sh 'python3 ./biblioteka_DL/dllib.py with "epochs=$EPOCHS"'
archiveArtifacts artifacts: 'model.pkl, s444018_sacred_FileObserver/**/*.*, result.csv', followSymlinks: false
sh 'python3 ./biblioteka_DL/dllib.py "epochs=$EPOCHS"'
archiveArtifacts artifacts: 'model.pkl, result.csv', followSymlinks: false
archiveArtifacts artifacts: 'mlruns/**'
archiveArtifacts artifacts: 'my_model/**'
build job: 's444018-evaluation/master/'

3
biblioteka_DL/dllib.py
View File

@ -15,8 +15,7 @@ from urllib.parse import urlparse
# mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s444018")
epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20
epochs = sys.argv[1]
def drop_relevant_columns(imbd_data):