upd51

2023-05-14 22:12:28 +02:00 · 2023-05-14 22:12:28 +02:00 · 9d0113448c
commit 9d0113448c
parent 56cfff9e5b
1 changed files with 2 additions and 1 deletions
--- a/script5_1.py
+++ b/script5_1.py
@ -55,7 +55,8 @@ y_train.fillna(y_train.mean(), inplace=True)
 # Convert the 'y_train' series to a NumPy array
 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
 model = Sequential()