Training a model from 2D Data