Update train.py

[dataubc]

trafficstars

The order of y_true, y_pred is reversed sklearn.metrics.mean_squared_error(y_true, y_pred, *, sample_weight=None, multioutput='uniform_average', squared=True)