sigmoid函数和阶跃函数的Python代码比较 ← 斋藤康毅

sigmoid函数就像“水车”一样，根据流过来的水量相应地调整传送出去的水量。
阶跃函数就像“竹筒敲石”一样，只做是否传递水（0或1）两个操作。

  

 备注：
“水车”图片来源于：https://new.qq.com/rain/a/20200520A04PLS00?pc
“竹筒敲石”图片来源于：https://www.sohu.com/a/207830556_503535

【sigmoid函数和阶跃函数的Python代码比较】

import numpy as np
import matplotlib.pylab as plt

def sigmoid(x):
    return 1/(1+np.exp(-x))    

def step_function(x):
    return np.array(x>0, dtype=np.int)

x=np.arange(-5.0, 5.0, 0.1)
y1=sigmoid(x)
y2=step_function(x)

plt.plot(x, y1,'c-') #青色实线
plt.plot(x, y2, 'r--') #红色虚线
plt.ylim(-0.1, 1.1)
plt.show()

【sigmoid函数和阶跃函数的图形】