深度学习入门——神经网络激活函数之阶跃函数

import numpy as np
import matplotlib.pyplot as plt


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


x = np.arange(-0.5, 0.5, 0.01)
y = step_function(x)

plt.plot(x, y)
plt.ylim(-0.1, 1.1) # 指定y轴的范围
plt.ylabel('y')
plt.xlabel('x')
plt.title('step function')
plt.show()