From e79bef16bff071c13e181fffaef1d6b7d24d59c9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Wed, 19 Apr 2023 17:08:19 +0200 Subject: [PATCH] Update 'create-dataset.py' --- create-dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/create-dataset.py b/create-dataset.py index 1303fcf..d252a23 100644 --- a/create-dataset.py +++ b/create-dataset.py @@ -22,9 +22,9 @@ X, Y = video_games, video_games X_train, X_temp, Y_train, Y_temp = train_test_split(X, Y, test_size=0.3, random_state=1) X_dev, X_test, Y_dev, Y_test = train_test_split(X_temp, Y_temp, test_size=0.3, random_state=1) -#X_train.to_csv('X_train.csv', index=False) -#X_dev.to_csv('X_dev.csv', index=False) -X.to_csv('X_test.csv', index=False) +X_train.to_csv('X_train.csv', index=False) +X_dev.to_csv('X_dev.csv', index=False) +X_test.to_csv('X_test.csv', index=False)