This commit is contained in:
Yevhenii Poliakov 2023-05-14 21:52:31 +02:00
parent 6caaed412f
commit f5693b54cc

View File

@ -51,6 +51,8 @@ model.compile(optimizer=Adam(), loss='mse')
print("Data type of 'Rating' column:", y_train.dtype) print("Data type of 'Rating' column:", y_train.dtype)
print("First few rows of 'y_train':", y_train[:10])
# Train the model # Train the model
model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test)) model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test))