I’m not sure I understand your suggestion; the individual model runs are already wrapped in the fit() function. Do you mean the entire process of constructing the models/optimizers/etc should be wrapped in some function call? That would seem to obfuscate the code a bit, given the amount of prep before training.