Intro import tensorflow as tf # Set up input data x = tf.range(100) # Set up neurons in hidden layer neurons = 5 # Number of neurons in the hidden layer activation = tf.keras.activations.relu # Activation function for the hidden layer # Define the model def neural_network_model(x, neurons, activation): # Define first layer first_layer = tf.keras.layers.Dense(units=neurons, activation=activation)