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

4
.gitignore vendored
View File

@ -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

7
Jenkinsfile vendored
View File

@ -23,4 +23,11 @@ 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)
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")
test_data = pd.read_csv("./test.csv")
train_data = pd.read_csv("./MoviesOnStreamingPlatforms_updated.train", header=None, usecols=cols)
# Stworzenie modelu
columns_to_use = ['Year', 'Runtime', 'Netflix']