Create a simple neural network

Use it directly keras.Model Method

def line_fit_model():
    """ Model Build a network structure  param: return: """
    #  Input layer 
    inputs = tf.keras.Input(shape=(1, ), name="inputs")
    #  Hidden layer -1
    layer1 = layers.Dense(10, activation="relu", name="layer1")(inputs)
    #  Hidden layer -2
    layer2 = layers.Dense(15, activation="relu", name="layer2")(layer1)
    #  Output layer 
    outputs = layers.Dense(5, activation="softmax", name="outputs")(layer2)
    #  Instantiation 
    model = tf.keras.Model(inputs=inputs, outputs=outputs)
    #  Show network structure 
    model.summary()
    return model

Inherit keras.Model Method

class LiftModel(tf.keras.Model):
    """  Inherit keras.Model  Rewrite calling function  """
    def __init__(self):
        super(LiftModel, self).__init__()
        self.layer1 = layers.Dense(10, activation=tf.nn.relu, name="layer1")
        self.layer2 = layers.Dense(15, activation=tf.nn.relu, name="layer2")
        self.outputs = layers.Dense(5, activation=tf.nn.softmax, name="outputs")
    def call(self, inputs):
        layer1 = self.layer1(inputs)
        layer2 = self.layer2(layer1)
        outputs = self.outputs(layer2)
        return outputs
if __name__ =="__main__":
	inputs = tf.constant([[1]])
	lift = LiftModel()
	lift(inputs)
	lift.summary()

use keras.Sequential Built in method

def line_fit_sequetnial():
    model = tf.keras.Sequential([
        layers.Dense(10, activation="relu", input_shape=(1, ),  name="layer1"),
        layers.Dense(15, activation="relu", name="layer2"),
        layers.Dense(5, activation="softmax", name="outputs")
    ])
    model.summary()
    return model

use Sequential() Outsourcing method

def outline_fit_sequential():
    model = tf.keras.Sequential()
    model.add(layers.Dense(10, activation="relu", input_shape=(1, ), name="layer1"))
    model.add(layers.Dense(15, activation="relu", name="layer2"))
    model.add(layers.Dense(5, activation="softmax", name="output"))
    model.summary()
    return model

Create a convolutional neural network

Built in Sequentia Method

def cnn_sequential():
    model = tf.keras.Sequential([
        #  Convolution layer -1
        layers.Conv2D(32, (3, 3), activation="relu", input_shape=(28, 28, 3), name="conv-1"),
        #  Pooling layer -1
        layers.MaxPooling2D((2, 2), name="pool-1"),
        #  Convolution layer -2
        layers.Conv2D(64, (3, 3),activation="relu" ,name="conv-2"),
        #  Pooling layer -2
        layers.MaxPooling2D((2, 2), name="pool-2"),
        #  Convolution layer -3
        layers.Conv2D(64, (3, 3), activation="relu", name="conv-3"),
        #  Flatten the column vector 
        layers.Flatten(),
        #  Fully connected layer -1
        layers.Dense(64, activation="relu", name="full-1"),
        # softmax layer 
        layers.Dense(64, activation="softmax", name="softmax-1")
    ])
    model.summary()

use Sequential Outsourcing method

def outline_cnn_sequential():
    model = tf.keras.Sequential()
    model.add(layers.Conv2D(32, (3, 3), activation="relu", input_shape=(84, 84, 3), name="conv-1" ))
    model.add(layers.MaxPooling2D((2, 2), name="pool-1"))
    model.add(layers.Conv2D(64, (3, 3), activation="relu", name="conv-2"))
    model.add(layers.MaxPooling2D((2, 2), name="pool-2"))
    model.add(layers.Conv2D(64, (3, 3), activation="relu", name="conv-3"))
    model.add(layers.MaxPooling2D((2, 2), name="pool-3"))
    model.add(layers.Flatten())
    model.add(layers.Dense(64, activation="relu", name='full-1'))
    model.add(layers.Dense(64, activation="softmax", name="softmax-1"))
    model.summary()