Congestion forecasting using a custom loss function