s487197/ium_487197
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Wojciech Lidwin 2023-05-12 04:46:05 +02:00
parent cdf8523c94
commit 4b806ee498
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Jenkinsfile_train
View File

@ -25,7 +25,7 @@ node {
checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
}
stage('Dockerfile'){
def testImage = docker.image('s487197/ium:49')
def testImage = docker.image('s487197/ium:50')
testImage.inside{
copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
sh "python3 ium_sacred.py -epochs $EPOCHS -lr $LR -validation_split $VALIDATION_SPLIT"

6
ium_sacred.py
View File

@ -49,11 +49,13 @@ def my_config():
lr = args.lr
validation_split = args.validation_split
@ex.capture
def prepare_message(epochs, lr, validation_split):
return "{0} {1} {2}!".format(epochs, lr, validation_split)
@ex.automain
@ex.main
def my_main(epochs, lr, validation_split, _run):
train = pd.read_csv('baltimore_train.csv')
@ -109,5 +111,5 @@ def my_main(epochs, lr, validation_split, _run):
_run.log_scalar('rmse', data['rmse'])
_run.log_scalar('accuracy', data['accuracy'])
"""
#ex.run()
ex.run()