From d1da0a938cae4debe3f79b871d8e6d76a9b800a5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Thu, 11 May 2023 19:32:06 +0200 Subject: [PATCH] Update 'create-dataset.py' --- create-dataset.py | 22 ---------------------- 1 file changed, 22 deletions(-) diff --git a/create-dataset.py b/create-dataset.py index 27d048b..5aa3e90 100644 --- a/create-dataset.py +++ b/create-dataset.py @@ -35,25 +35,3 @@ X_test.to_csv('X_test.csv', index=False) Y_test.to_csv('Y_test.csv', index=False) Y_train.to_csv('Y_train.csv', index=False) Y_dev.to_csv('Y_dev.csv', index=False) - -train_data_x = pandas.read_csv('./X_train.csv') - - -games_all = train_data_x.copy() -games_predict = train_data_x.pop('User_Score') -normalize = layers.Normalization() -normalize.adapt(games_all) - -norm_games_model = tensorflow.keras.Sequential([ - normalize, - layers.Dense(64), - layers.Dense(1) -]) - -norm_games_model.compile( - loss=tensorflow.keras.losses.MeanSquaredError(), - optimizer=tensorflow.keras.optimizers.Adam()) - -norm_games_model.fit(games_all, games_predict, epochs=500) - -norm_games_model.save('test')