Learning optimization solutions