s434704/ium_434704
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Training script fixes

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-05-14 00:20:04 +02:00
parent 192184cb2c
commit 999da6af32
4 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.gitignore vendored
View File

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

9
Jenkinsfile vendored
View File

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

7
Jenkinsfile_stats
View File

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

5
training.py
View File

@ -9,9 +9,10 @@ from tensorflow.keras.layers.experimental import preprocessing
pd.set_option("display.max_columns", None) 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 # 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 # Stworzenie modelu
columns_to_use = ['Year', 'Runtime', 'Netflix'] columns_to_use = ['Year', 'Runtime', 'Netflix']