tf_building_network.py

class SingleHiddenLayerNetwork(tf.keras.Model):
  def __init__(self):
    super(SingleHiddenLayerNetwork, self).__init__()
    self.hidden_layer = tf.layers.Dense(10, activation=tf.nn.tanh, use_bias=True)
    self.output_layer = tf.layers.Dense(3, use_bias=True, activation=None)

  def call(self, x):
    # Forward-pass logic
    return self.output_layer(self.hidden_layer(x))