【Matplotlib绘制激活函数】

import matplotlib.pyplot as plt
import numpy as np
from matplotlib import rcParams

plt.figure(figsize=(8, 6))
config = {
    "font.family": 'Times New Roman',
    "font.size": 16,
    "mathtext.fontset": 'stix',
    "font.serif": ['Times New Roman'],
}
rcParams.update(config)

x = np.linspace(-3, 1, 400)
plt.plot([0, 0], [1, -1], c="k", ls=":", lw=1.5)
plt.plot([-3, 1], [0, 0], c="k", ls=":", lw=1.5)

y_1_mish = x * np.tanh(np.log(1 + np.exp(x)))
plt.plot(x, y_1_mish, label="Mish")

y_2_ReLU = np.maximum(0, x)
plt.plot(x, y_2_ReLU,label="ReLU")

y_3_Leaky_ReLU = np.maximum(0.1 * x, x)
plt.plot(x, y_3_Leaky_ReLU,label="Leaky ReLU")

plt.xlabel(r"$/x$", fontsize=20)
plt.ylabel(r"$/y$", fontsize=20)

plt.xlim(-3, 1)
plt.ylim(-0.32, 1)
plt.legend(fontsize=16)
plt.xticks(fontsize=16)
plt.yticks(fontsize=16)

plt.savefig("figure.tiff", dpi=300, bbox_inches="tight")
plt.show()