This commit is contained in:
Yevhenii Poliakov 2023-05-14 22:14:39 +02:00
parent 9d0113448c
commit c3b7f91d57

View File

@ -56,7 +56,8 @@ y_train.fillna(y_train.mean(), inplace=True)
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)])
y_train = y_train.astype(float)
y_train = y_train[np.isfinite(y_train)]
# Create the neural network model
model = Sequential()