upd51
This commit is contained in:
parent
9d0113448c
commit
c3b7f91d57
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user