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("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))
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user