adding trainparams full model as documentation