This commit is contained in:
Yevhenii Poliakov 2023-05-14 22:21:18 +02:00
parent 96b97aca98
commit ea7ff238f7

View File

@ -75,6 +75,11 @@ print("Data type of 'Rating' column:", y_train.dtype)
print("First few rows of 'y_train':", y_train[:10])
print("Data type of 'X_train':", X_train.dtype)
print("Shape of 'X_train':", X_train.shape)
print("Data type of 'y_train':", y_train.dtype)
print("Shape of 'y_train':", y_train.shape)
# Train the model
model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test))