Python绘制三种常见的激活函数

import numpy as np
import matplotlib.pyplot as plt
# 定义sigmoid函数
def sigmoid(x):
return 1 / (1 + np.exp(-x))
# 定义ReLU函数
def relu(x):
return np.maximum(0, x)
# 定义tanh函数
def tanh(x):
return np.tanh(x)
# 生成x轴数据
x = np.linspace(-10, 10, 1000)
# 绘制sigmoid函数
y_sigmoid = sigmoid(x)
plt.plot(x, y_sigmoid, label='sigmoid')
# 绘制ReLU函数
y_relu = relu(x)
plt.plot(x, y_relu, label='ReLU')
# 绘制tanh函数
y_tanh = tanh(x)
plt.plot(x, y_tanh, label='tanh')
# 添加图例和标题
plt.legend()
plt.title('Three Common Activation Functions')
plt.xlabel('x')
plt.ylabel('y')
plt.grid(True)
plt.show()