From 9af96b100ccb32f18a7a275ba8c06936f542ee54 Mon Sep 17 00:00:00 2001 From: Karolina Oparczyk Date: Mon, 17 May 2021 20:40:08 +0200 Subject: [PATCH] rmse --- JenkinsfileNeural | 2 +- neural_network.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/JenkinsfileNeural b/JenkinsfileNeural index 02c8809..b42fa9c 100644 --- a/JenkinsfileNeural +++ b/JenkinsfileNeural @@ -12,7 +12,7 @@ node { } stage('Clone repo') { - docker.image("karopa/ium:09").inside { + docker.image("karopa/ium:10").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") diff --git a/neural_network.py b/neural_network.py index e193edd..07de086 100644 --- a/neural_network.py +++ b/neural_network.py @@ -39,7 +39,7 @@ model.compile(loss='mean_absolute_error', optimizer="Adam", metrics=['mean_absol model.fit(X, y, epochs=30, validation_split = 0.3) -data = pd.read_csv("data_dev", sep=',', error_bad_lines=False, skip_blank_lines=True).dropna() +data = pd.read_csv("data_dev", sep=',', error_bad_lines=False, skip_blank_lines=True, nrows=527).dropna() X_test = data.loc[:,data.columns == "440265"].astype(int) y_test = data.loc[:,data.columns == "21629"].astype(int)