diff --git a/script5_1.py b/script5_1.py
index 4d88b03..df574ab 100644
--- 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))