신경망 - 계단 함수, 계단 함수 그래프 구현

728x90

계단 함수

파이썬 코드

import numpy as np
x = np.array([-1.0, 1.0, 2.0])
print(x)

y = x>0
print(y) 
# x의 원소 각각이 0보다 크면 True, 0이하면 False로 변환한 새로운 배열 y가 생성된다.
# 이렇게 만들어진 y는 bool배열이기 때문에 int형으로 바꿔준다.

y = y.astype(np.int)
print(y)

# 넘파이 배열의 자료형 변환은 astype()

계단 함수의 그래프

파이썬 코드

# 계단 함수 그래프 

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) # y축의 범위를 지정
plt.show()