federated/models/models.py
def create_softmax_model() -> tf.keras.Sequential:
    """Creates a softmax regression model.

    Returns:
        tf.keras.Sequential: A softmax regresion model.
    """
    model = Sequential(
        [
            layers.InputLayer(input_shape=[186, 1]),
            layers.Flatten(),
            layers.Dense(5),
        ]
    )
    return model
Creates a softmax regression model.

    Returns:
        tf.keras.Sequential: A softmax regresion model.
    
Creates a softmax regression model.
Creates a softmax regression model .
def create_softmax_model() -> tf.keras.Sequential:
    model = Sequential(
        [
            layers.InputLayer(input_shape=[186, 1]),
            layers.Flatten(),
            layers.Dense(5),
        ]
    )
    return model
def create_softmax_model ( ) -> tf . keras . Sequential : model = Sequential ( [ layers . InputLayer ( input_shape = [ 186 , 1 ] ) , layers . Flatten ( ) , layers . Dense ( 5 ) , ] ) return model
Creates a softmax regression model.
Creates a softmax regression model .
"""Creates a softmax regression model.

    Returns:
        tf.keras.Sequential: A softmax regresion model.
    """
A softmax regresion model.
A softmax regresion model .
tf.keras.Sequential