This commit is contained in:
Yevhenii Poliakov 2023-05-14 22:12:28 +02:00
parent 56cfff9e5b
commit 9d0113448c

View File

@ -55,7 +55,8 @@ y_train.fillna(y_train.mean(), inplace=True)
# Convert the 'y_train' series to a NumPy array # Convert the 'y_train' series to a NumPy array
y_train = y_train.values y_train = y_train.values
# Filter out non-numeric and NaN values from y_train
y_train = np.array([x for x in y_train if np.isfinite(x)])
# Create the neural network model # Create the neural network model
model = Sequential() model = Sequential()