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

Compare commits

base: s487197:90ea83a9f0e6b2d2870d4c21c3b24de1cb71d80f
Branches Tags
s487197:master
...
compare: s487197:79f0ab18db8e9c2f0e6bee0469561a4f90117b47
Branches Tags
s487197:master

2 Commits
90ea83a9f0 ... 79f0ab18db

Author SHA1 Message Date
Wojciech Lidwin
79f0ab18db Fix 2023-05-11 19:57:26 +02:00
Wojciech Lidwin
f1f3d2647a Model fix 2023-05-11 19:45:43 +02:00
4 changed files with 4 additions and 3 deletions
Show Stats Expand all files Collapse all files

1
Dockerfile
View File

@ -1,6 +1,7 @@
FROM ubuntu:latest FROM ubuntu:latest
RUN apt update && apt install -y figlet RUN apt update && apt install -y figlet
RUN apt install unzip -y
RUN apt install python3-pip -y RUN apt install python3-pip -y
RUN pip3 install kaggle RUN pip3 install kaggle

2
Jenkinsfile_train
View File

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

2
ium_predict.py
View File

@ -26,7 +26,7 @@ def get_x_y(data):
def predict(): def predict():
model = load_model('baltimore_model3') model = load_model('baltimore_model.h5')
train = pd.read_csv('baltimore_train.csv') train = pd.read_csv('baltimore_train.csv')
baltimore_data_test = pd.read_csv('baltimore_test.csv') baltimore_data_test = pd.read_csv('baltimore_test.csv')

2
ium_train.py
View File

@ -48,7 +48,7 @@ def train_model():
validation_split=0.2) validation_split=0.2)
hist = pd.DataFrame(history.history) hist = pd.DataFrame(history.history)
hist['epoch'] = history.epoch hist['epoch'] = history.epoch
model.save('baltimore_model3') model.save('baltimore_model.h5')
train_model() train_model()