Custom model training user interface