Build keras model

buildModelKeras.py

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

def build_model(learning_rate, layer_size):
    """Build Keras model"""
    model = keras.Sequential([
      layers.Dense(layer_size, activation='relu',
                   input_shape=[len(feature_names)]),
      layers.Dense(layer_size, activation='relu', kernel_regularizer='l2'),
      layers.Dense(1, activation='sigmoid')
    ])
    optimizer = tf.keras.optimizers.Adam(learning_rate=learning_rate)

    model.compile(loss='mse', optimizer=optimizer, metrics=['mae', 'mse'])
    return model
  
  # Later called by:
  
  regressor_model = build_model(lr, layer_Size)