s434686/ium_z434686
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'create-dataset.py'

Browse Source
This commit is contained in:
Patryk Gałka 2023-05-11 19:32:06 +02:00
parent 5077335302
commit d1da0a938c
1 changed files with 0 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
create-dataset.py
View File

@ -35,25 +35,3 @@ X_test.to_csv('X_test.csv', index=False)
Y_test.to_csv('Y_test.csv', index=False) Y_test.to_csv('Y_test.csv', index=False)
Y_train.to_csv('Y_train.csv', index=False) Y_train.to_csv('Y_train.csv', index=False)
Y_dev.to_csv('Y_dev.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')