y = x > 0 # 들어온 배열에서 0보다 큰 경우만 골라 1을 반환한다.(True 반환)
y = x > 0 # 들어온 배열에서 0보다 큰 경우만 골라 1을 반환한다.(True 반환)
return y.astype(np.int) # bool형을 반환하기 때문에 int형으로 바꾸어준다.
return y.astype(np.int) # bool형을 반환하기 때문에 int형으로 바꾸어준다.
−
</syntaxhighlight>
+
</syntaxhighlight>계단함수는 잘 사용되지 않는데, 그 이유는 훈련이 진행됨에 따라(매개변수가 변함에 따라) 정확도나 손실함수가 매끄럽지 않고 불연속적으로 갑작스레 변화한다. 이렇게 되면 학습이 잘 이루어지고 있는지 확인하기 어렵다. 때문에 계단함수는 잘 사용되지 않는다.