network.py

import tensorflow as tf
from tf.keras.regularizers import l2

def build_nn(input_shape, 
             hidden_units=(20, 20), 
             output_size=1, 
             output_activation=None,
             hidden_activation=tf.nn.relu, 
             dropout=0.0, 
             regularization=0.0):
    inputs = tf.keras.Input(shape=input_shape)
    
    hidden = inputs
    for layer_units in hidden_units:
        hidden = tf.keras.layers.Dense(layer_units, activation=hidden_activation, kernel_regularizer=l2(regularization))(hidden)
        hidden = tf.keras.lyaers.Dropout(dropout)(hidden)
        
    outputs = tf.keras.layers.Dense(output_size, activation=output_activation, kernel_regularizer=l2(regularization))(hidden)
    
    model = tf.keras.Model(inputs=inputs, outputs=outputs)
    return model