[머신러닝 - Python] 활성화 함수 - 계단 함수 구현 (Activation Function - Step Function Implementation)

계단 함수는 0이하의 값에 대해 0을 출력하고, 0초과의 값에 대해서는 1을 출력한다.
numpy 배열에 대해서 각각 계산을 할 수 있게 만들면 코드는 아래와 같다.

import numpy as np
import matplotlib.pyplot as plt


def step(x):
    return np.array(x > 0, dtype=np.int64)


x = np.arange(-5.0, 5.0, 0.1)
y = step(x)
plt.plot(x, y)
plt.ylim(-0.1, 1.1)
plt.show()

실제로 -5.0 부터 5.0까지 0.1 간격으로 벡터를 만들어서 넣었는데, 이런 형태의 함수가 나오게 된다.
참고로 x > 0만 하면 True, False 값으로 나오는데 이를 dtype = np.int64를 해줌으로써 bool 형태를 int 형태로 바꿔준다.



감사합니다.


지적 환영합니다.