Train NN Keras

trainNNKeras.py

def train_model(model, feature, label, epochs, batch_size):
    """Train Keras Model"""
    history = model.fit(x=feature,
                        y=label,
                        batch_size=batch_size,
                        epochs=epochs)
    epochs = history.epoch
    hist = pd.DataFrame(history.history)
    mse = hist["mse"]
    return epochs, mse
  
  epochs, rmse = train_model(regressor_model, train_Features, train_Target, epochs, batch_size)