fuel_model.py

model = models.Sequential([
    layers.Dense(1024, activation = 'relu', input_shape = Xtrain[0].shape),
    layers.Dense(512, activation = 'relu'),
    layers.Dense(256, activation = 'relu'),
    layers.Dense(64, activation = 'relu'),
    layers.Dense(16, activation = 'relu'),
    layers.Dense(1)
])

cb = callbacks.EarlyStopping(patience = 20, restore_best_weights = True)

model.compile(optimizer = optimizers.Adam(0.00001), loss = losses.MeanSquaredError(), metrics = ['mae'])
history = model.fit(Xtrain, ytrain, validation_data = (Xval, yval), epochs = 256, callbacks = cb)