当前位置：网站首页>Step function of activation function

Step function of activation function

2022-04-23 11:40:00 【Hey, little black 15kg】

Step function

The step function is expressed in 0 As a boundary , Output from 0 Switch to a 1（ Or from 1 Switch to a 0）. Its value changes stepwise , So it's called step function .

import matplotlib.pyplot as plt
import numpy as np

def step_function(x):
    y = x>0
    print(y.shape,y.dtype)
    return y.astype(np.int32)

# def step_function(x):
# print(x>0)
# return np.array(x>0,dtype=np.int32)


if __name__ == '__main__':

    x = np.arange(-5,5,0.1)
    y = step_function(x)

    plt.plot(x,y)
    plt.ylim(-0.1,1.1)
    plt.show()