asserts for feature dimension
seperate creation of model and training of model