Code to reproduce figure 3 about entmax HOT 1 CLOSED

t = np.linspace(-3.0, 3.0, 100)
x = [[i, 0.0] for i in t]
x = torch.tensor(x)
softmax_y = softmax(x)
alpha125_y = entmax_bisect(x, 1.25)
alpha15_y = entmax_bisect(x, 1.5)
sparsemax_y = sparsemax(x)
alpha4_y = entmax_bisect(x, 4)

fig, ax = plt.subplots(figsize=(8, 6))
ax.plot(t, softmax_y[:, 0], label=r"$\alpha = 1$ (softmax)", linewidth=3, linestyle=':')
ax.plot(t, alpha125_y[:, 0], label=r"$\alpha = 1.25$", linewidth=2)
ax.plot(t, alpha15_y[:, 0], label=r"$\alpha = 1.5$", linewidth=3, linestyle='--')
ax.plot(t, sparsemax_y[:, 0], label=r"$\alpha = 2$ (sparsemax)", linewidth=3)
ax.plot(t, alpha4_y[:, 0], label=r"$\alpha = 4$", linewidth=1, color='k')
ax.set_yticks([0.0, 0.5, 1.0])
plt.legend()