This commit is contained in:
Yevhenii Poliakov 2023-05-14 22:06:04 +02:00
parent 220c71a812
commit 2533663657

View File

@ -44,9 +44,10 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
# Convert the modified 'Rating' column to a numpy array
y_train_updated = data['Rating'].to_numpy()
y_train_updated = pd.Series(y_train_updated[:len(X_train)])
# Update the 'y_train' array with the modified values
y_train.loc[X_train.index] = y_train_updated[:len(X_train)]
y_train.loc[X_train.index] = y_train_updated
# Create the neural network model
model = Sequential()