激活函数图绘制Python代码

最新推荐文章于 2024-01-08 17:19:44 发布

周星星✘

最新推荐文章于 2024-01-08 17:19:44 发布

阅读量349

点赞数

文章标签： 1024程序员节 python 深度学习机器学习

本文链接：https://blog.csdn.net/qq_45152711/article/details/127499213

版权

该博客展示了如何使用Python的numpy和matplotlib库绘制Sigmoid和ReLU激活函数的图形。代码分别计算了两种函数的值，并将结果以图像形式保存，便于理解这两种在神经网络中常用的非线性变换。

摘要由CSDN通过智能技术生成

# -- coding: utf-8 --
import numpy as np
import math
import matplotlib.pyplot as plt
x = np.arange(-10, 10, 0.1)
y = []
title_name='Sigmoid'
for t in x:
    if title_name=='Sigmoid':
        y_1 = 1 / (1 + math.exp(-t))
    if title_name=='ReLU':
        if t<0:
            y_1=0
        else:
            y_1=t
    y.append(y_1)
plt.title(title_name, fontsize=12)
plt.tick_params(axis='both', which='major', labelsize=8)
plt.plot(x, y, label="sigmoid")
plt.xlabel("x",fontsize=12)
plt.ylabel("f(x)",fontsize=12)
# plt.ylim(0, 1)
# plt.legend()
plt.tight_layout()
plt.savefig('./'+title_name+'.png', pad_inches=0, dpi=600)
plt.show()