Rail traction energy management model