upd51

2023-05-14 21:50:30 +02:00 · 2023-05-14 21:50:30 +02:00 · 6caaed412f
commit 6caaed412f
parent af22e68a9f
1 changed files with 2 additions and 0 deletions
--- a/script5_1.py
+++ b/script5_1.py
@ -49,6 +49,8 @@ model.add(Dense(1))
 # Compile the model
 model.compile(optimizer=Adam(), loss='mse')

+print("Data type of 'Rating' column:", y_train.dtype)
+
 # Train the model
 model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test))