flexs.baselines.models.keras_model

Define the base KerasModel class.

class flexs.baselines.models.keras_model.KerasModel(model, alphabet, name, batch_size=256, epochs=20, custom_train_function=None, custom_predict_function=None)

Bases: flexs.model.Model

A wrapper around tensorflow/keras models.

train(sequences, labels, verbose=False)

Train keras model.