flexs.baselines.models.cnn

Define a baseline CNN Model.

class flexs.baselines.models.cnn.CNN(seq_len, num_filters, hidden_size, alphabet, loss='MSE', name=None, batch_size=256, epochs=20)[source]

Bases: flexs.baselines.models.keras_model.KerasModel

A baseline CNN model with 3 conv layers and 2 dense layers.