Supongamos que partimos del siguiente dataset:
df = pd.DataFrame({
"x0": [0.3, 0.1, 0.4, 0.5],
"x1": [0.7, 0.2, 0.3, 0.8],
"y": [1, 0, 0, 1]
})
df
En x0 y x1 tenemos las características predictivas y en y la variable objetivo (la etiqueta), que vamos a considerar categórica para trabajar en un escenario de clasificación. Como es habitual, creamos los conjuntos con las características predictivas y la etiqueta:
y_train = df.pop("y")
X_train = df
El objetivo es entrenar una red neuronal con una capa oculta de 2 neuronas.
