계단 함수(Step Function)

1. 계단 함수

 

계단 함수는 아래의 식과 같이 입력이 0을 넘으면  1을 출력하고, 그 외에는 0을 출력하는 함수이다.

 

h(x) = 0 (x<= 0)
     = 1 (x > 0)

 

1.1 계단 함수 구현

 

step_function.py

# coding: utf-8
import numpy as np
import matplotlib.pylab as plt

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

X = np.arange(-5.0, 5.0, 0.1) # -5.0에서 5.0 전까지 0.1 간격의 넘파이 배열을 생성
Y = step_function(X) # 계단 함수 실행 결과는 배열로 리턴
plt.plot(X, Y)
plt.ylim(-0.1, 1.1)
plt.show()

 

 

1.2 계단 함수의 그래프

 

 

Github: https://github.com/hwauni/MachineLearning/tree/master/ML_alg/StepFunction