Training script fixes

.gitignore

@@ -150,3 +150,7 @@ Thumbs.db #thumbnail cache on Windows
 
 env
 # End of https://www.toptal.com/developers/gitignore/api/python
+
+/MoviesOnStreamingPlatforms_updated.train
+/MoviesOnStreamingPlatforms_updated.test
+/MoviesOnStreamingPlatforms_updated.dev

Jenkinsfile

@@ -23,4 +23,11 @@ stages {
         }
     }
     }
+     post {
+      success {
+         build job: 's434704-training/master', parameters: [
+                string(name: 'options', value: "-shuffle=true")
+                ]
+      }
+   }       
 }

Jenkinsfile_stats

@@ -19,11 +19,4 @@ pipeline {
             }
         }
     }
-    post {
-      success {
-         build job: 's434704-training/master', parameters: [
-                string(name: 'options', value: "-shuffle=true")
-                ]
-      }
-   }
 }

training.py

@@ -9,9 +9,10 @@ from tensorflow.keras.layers.experimental import preprocessing
 
 pd.set_option("display.max_columns", None)
 
+cols = ['0','ID','Title','Year','Age','IMDb','Rotten Tomatoes','Netflix','Hulu','Prime Video','Disney+','Type','Directors','Genres','Country','Language','Runtime']
+
 # Wczytanie danych
-train_data = pd.read_csv("./train.csv")
+train_data = pd.read_csv("./MoviesOnStreamingPlatforms_updated.train", header=None, usecols=cols)
-test_data = pd.read_csv("./test.csv")
 
 # Stworzenie modelu 
 columns_to_use = ['Year', 'Runtime', 'Netflix']