upd51
This commit is contained in:
parent
6caaed412f
commit
f5693b54cc
@ -51,6 +51,8 @@ model.compile(optimizer=Adam(), loss='mse')
|
||||
|
||||
print("Data type of 'Rating' column:", y_train.dtype)
|
||||
|
||||
print("First few rows of 'y_train':", y_train[:10])
|
||||
|
||||
# Train the model
|
||||
model.fit(X_train, y_train, batch_size=64, epochs=10, validation_data=(X_test, y_test))
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user