From e0ecd91f6d3afffc21b3057563fb37b1e09b2a1e Mon Sep 17 00:00:00 2001 From: Maciej Czajka Date: Mon, 21 Mar 2022 08:52:14 +0100 Subject: [PATCH] update create train_dev_test --- Zajęcia 2/main.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/Zajęcia 2/main.py b/Zajęcia 2/main.py index b004136..23b3cf6 100644 --- a/Zajęcia 2/main.py +++ b/Zajęcia 2/main.py @@ -66,9 +66,11 @@ def create_plot(games): plt.show() # funkcja dokonująca podziału na podzbiory train i test -def create_train_test(games): - games_train, games_test = sklearn.model_selection.train_test_split(games, test_size=5810, random_state=1) - return games_train, games_test +def create_train_dev_test(games): + games_train, games_test = sklearn.model_selection.train_test_split(games, test_size=6972, random_state=1) + games_test, games_dev = sklearn.model_selection.train_test_split(games_test, test_size=3486, random_state=1) + #games_train, games_test = sklearn.model_selection.train_test_split(games, test_size=5810, random_state=1) + return games_train, games_dev, games_test # funkcja pobierająca informacje z danej platformy o ocenach metacritic def get_platform_score_metacritic(games, platform): @@ -167,5 +169,6 @@ dataset_information(games) # wypisanie informacji o danej platformie platform_information(games, " Xbox 360") -# podział na podzbiory train i test -games_train, games_test = create_train_test(games) \ No newline at end of file +# podział na podzbiory train, dev i test (6,2,2) +#games_train, games_test = create_train_test(games) +games_train, games_dev, games_test = create_train_dev_test(games)