From b73b07faf1670fa3e06778c69940b7c7f18f831a Mon Sep 17 00:00:00 2001
From: Dominik Strzako <sylvaripl@gmail.com>
Date: Fri, 14 May 2021 19:47:01 +0200
Subject: [PATCH] :)

---
 Jenkinsfile_train      | 2 +-
 Zadanie_06_training.py | 6 +-----
 requirements.txt       | 9 ++++++---
 3 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 1761e37..8cea8f0 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -1,5 +1,5 @@
 pipeline {
-    agent {dockerfile true}
+    agent {image 'adnovac/ium_s434760:1.2'}
 	parameters{
         buildSelector(
             defaultSelector: lastSuccessful(),
diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py
index 0c082c8..c1aa885 100644
--- a/Zadanie_06_training.py
+++ b/Zadanie_06_training.py
@@ -3,14 +3,10 @@ from tensorflow.keras.layers import Dense
 from sklearn.metrics import accuracy_score, classification_report
 import pandas as pd
 from sklearn.model_selection import train_test_split
-import wget
 import numpy as np
 import os
 
-url = 'https://git.wmi.amu.edu.pl/s434788/ium_434788/raw/branch/master/winequality-red.csv'
-wget.download(url, out='winequality-red.csv', bar=None)
-
-wine=pd.read_csv('winequality-red.csv')
+wine=pd.read_csv('train.csv')
 wine
 
 y = wine.quality
diff --git a/requirements.txt b/requirements.txt
index 64cbc17..495e906 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,8 @@
+kaggle==1.5.12
 pandas
-numpy
+numpy==1.19.2
 sklearn
-tensorflow
-wget
+tensorflow==2.4.1
+jinja2==2.11.3
+matplotlib
+wget
\ No newline at end of file