绘制激活函数及其导数(pytorch)(基础版)

最新推荐文章于 2025-04-02 13:37:25 发布

艾叶白果

最新推荐文章于 2025-04-02 13:37:25 发布

阅读量3.3k

点赞数 7

分类专栏：机器学习深度学习文章标签：机器学习激活函数 pytorch matplotlib

本文链接：https://blog.youkuaiyun.com/huadushao/article/details/107094804

版权

基本的激活函数及其导数绘图（导数部分使用pytorch的自动求导）

本文包括的激活函数有：
1.sigmoid
2.tanh
3.relu
4.leakyrelu
5.elu

准备绘图函数

%matplotlib inline
import torch
import numpy as np
import matplotlib.pylab as plt
import sys
sys.path.append("..") 
from matplotlib import pyplot as plt
from IPython import display
from matplotlib import style

def xyplot(x_vals, y_vals, name):
    display.set_matplotlib_formats('svg')
    plt.rcParams['figure.figsize'] = (5, 3.5)
    plt.plot(x_vals.detach().numpy(), y_vals.detach().numpy(), label = name, linewidth=1.5, color='#FF0000')
    plt.grid(True,linestyle=':')   
    plt.legend(loc