Cambié la arquitectura de la red y 80 epochs:
def build_model_regression(input_data):
model = models.Sequential()
model.add(layers.Dense(50,activation='relu',input_shape=(input_data,)))
model.add(layers.Dropout(0.5))
model.add(layers.Dense(50,activation='relu'))
model.add(layers.Dropout(0.5))
model.add(layers.Dense(1))
model.compile(optimizer='rmsprop', loss='mse',metrics=['mae'])
return model
loss: 19.8648 - mae: 3.1667
Los aportes, preguntas y respuestas son vitales para aprender en comunidad. Regístrate o inicia sesión para participar.