cerebral.models.build_model(train_ds, num_shared_layers, num_specific_layers, units_per_layer, regularizer='l2', regularizer_rate=0.001, dropout=0.3, learning_rate=0.01, ensemble_size=1, activation='relu', max_norm=3)[source]

Constructs a model containing a shared branch, and individual feature branches. See Figure 1 of https://pubs.rsc.org/en/content/articlelanding/2022/dd/d2dd00026a.

Last update: Feb 09, 2023