models

models.models module

federated.models.models.create_cnn_model() → tensorflow.python.keras.engine.sequential.Sequential

Creates 1D CNN model.

Returns:

tf.keras.Sequential: A 1D CNN model.

federated.models.models.create_dense_model() → tensorflow.python.keras.engine.sequential.Sequential

Creates ANN model.

Returns:

tf.keras.Sequential: An ANN model.

federated.models.models.create_softmax_model() → tensorflow.python.keras.engine.sequential.Sequential

Creates a softmax regression model.

Returns:

tf.keras.Sequential: A softmax regresion model.