ax.plot(xi, yi, 'r', label='True Probability')