rmse

Dockerfile

@@ -7,6 +7,7 @@ RUN apt install -y unzip
 RUN pip3 install pandas
 RUN pip3 install kaggle
 RUN pip3 install tensorflow
+RUN pip3 install sklearn
 COPY ./data_train ./
 COPY ./data_dev ./
 COPY ./neural_network.sh ./

JenkinsfileNeural

@@ -12,7 +12,7 @@ node {
 
         }
         stage('Clone repo') {
-                    docker.image("karopa/ium:06").inside {
+                    docker.image("karopa/ium:07").inside {
                         stage('Test') {
                             checkout([$class: 'GitSCM', branches: [[name: '*/master']], doGenerateSubmoduleConfigurations: false, extensions: [], submoduleCfg: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s434765/ium_434765']]])
                             copyArtifacts fingerprintArtifacts: true, projectName: 's434765-create-dataset', selector: buildParameter("BUILD_SELECTOR")

neural_network.py

@@ -1,6 +1,5 @@
 import pandas as pd
 import numpy as np
-from keras import optimizers
 from sklearn.metrics import mean_squared_error
 
 from tensorflow import keras