I\'m trying to train a simple linear model, and for some reason, it keeps running into an error, as it doesn\'t like the shape of the data. Does anyone know the way to make this