python画激活函数图片
时间: 2023-07-21 19:59:51 浏览: 102
当然可以!以下是使用Python绘制常见的激活函数图像的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义激活函数
def sigmoid(x):
return 1 / (1 + np.exp(-x))
def relu(x):
return np.maximum(0, x)
def tanh(x):
return np.tanh(x)
def softmax(x):
exps = np.exp(x - np.max(x))
return exps / np.sum(exps)
# 定义输入范围
x = np.linspace(-10, 10, 100)
# 计算激活函数输出
y_sigmoid = sigmoid(x)
y_relu = relu(x)
y_tanh = tanh(x)
# 绘制激活函数图像
plt.figure(figsize=(10, 6))
plt.subplot(2, 2, 1)
plt.plot(x, y_sigmoid, label='Sigmoid')
plt.xlabel('x')
plt.ylabel('Sigmoid(x)')
plt.legend()
plt.subplot(2, 2, 2)
plt.plot(x, y_relu, label='ReLU')
plt.xlabel('x')
plt.ylabel('ReLU(x)')
plt.legend()
plt.subplot(2, 2, 3)
plt.plot(x, y_tanh, label='Tanh')
plt.xlabel('x')
plt.ylabel('Tanh(x)')
plt.legend()
plt.tight_layout()
plt.show()
```
这段代码会绘制出Sigmoid、ReLU和Tanh三种常见激活函数的图像。你可以根据需要修改代码中的激活函数或输入范围来绘制其他激活函数的图像。运行代码后,会显示一个包含三个子图的图像窗口,每个子图对应一个激活函数的图像。
阅读全文