upd51
This commit is contained in:
parent
96b97aca98
commit
ea7ff238f7
@ -75,6 +75,11 @@ print("Data type of 'Rating' column:", y_train.dtype)
|
|||||||
|
|
||||||
print("First few rows of 'y_train':", y_train[:10])
|
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
|
# 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