본문 바로가기
IT/머신러닝

[Keras] 기본 예제

by 빨강자몽 2018. 8. 31.

 츨처 : <"코딩셰프의 3분 딥러닝 케라스맛" 김성진 저>를 공부하면서 작성하였습니다.



코드


1개의 입력(x)에 대해서 1개의 출력(y)값을 예측해보는 기본 모델

# basic linearRegression by karas
# 패키지 임포트
import keras
import numpy

# 입력
x = numpy.array([0,1,2,3,4])
y = x * 2 + 1

# 모델 구성
model = keras.models.Sequential()
model.add(keras.layers.Dense(1,input_shape=(1,)))
model.compile('SGD','mse')

# epoch 세대, verbose 학습 진행 사항 표시 여부(0 표시 않함)
model.fit(x[:2],y[:2],epochs=1000,verbose=0)

# 출력 결과 확인
print('Target',y[2:])
print('Predictions:',model.predict(x[2:]).flatten())

실행 결과