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-11 20:46:34 +02:00
parent 64c093204d
commit 0735fbc4ef
2 changed files with 2 additions and 2 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:35')
def testImage = docker.image('s487197/ium:36')
testImage.inside{
copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
sh "python3 ium_train.py -epochs $EPOCHS -lr $LR -validation_split $VALIDATION_SPLIT"

2
ium_train.py
View File

@ -33,7 +33,7 @@ def train_model():
parser.add_argument('-epochs', type=int, default=20)
parser.add_argument('-lr', type=float, default=0.01)
parser.add_argument('-validation_split', type=int, default=0.2)
parser.add_argument('-validation_split', type=float, default=0.2)
args = parser.parse_args()
train = pd.read_csv('baltimore_train.csv')