This commit is contained in:
Yevhenii Poliakov 2023-05-14 22:03:43 +02:00
parent caf64dc058
commit 220c71a812

View File

@ -46,7 +46,7 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
y_train_updated = data['Rating'].to_numpy() y_train_updated = data['Rating'].to_numpy()
# Update the 'y_train' array with the modified values # Update the 'y_train' array with the modified values
y_train.loc[X_train.index] = y_train_updated y_train.loc[X_train.index] = y_train_updated[:len(X_train)]
# Create the neural network model # Create the neural network model
model = Sequential() model = Sequential()