This commit is contained in:
Yevhenii Poliakov 2023-05-14 22:00:58 +02:00
parent e2724d9ee1
commit caf64dc058

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 = y_train_updated y_train.loc[X_train.index] = y_train_updated
# Create the neural network model # Create the neural network model
model = Sequential() model = Sequential()