【Matplotlib绘制激活函数】

Matplotlib绘制函数图像（Mish/ReLU/Leaky ReLU激活函数）

导入模块

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)

Mish

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

ReLU

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

Leaky_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()