파이썬, 단순 선형 회귀 예제(텐서플로우2)

단순 선형 회귀 예제 (텐서플로우2)

 

 

글. 오상문 sualchi@daum.net

 

# 단순한 선형 회귀 예제 (텐서플로우2)
import numpy as np
import tensorflow as tf

 

# x(입력), y(결과) 데이터
x_train = [1,2,3,4]
y_train = [1,2,3,4]

 

# 모델 생성
model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Dense(1, input_dim=1)) # 1차원 입력자료

 

# 최적화 방식
sgd = tf.keras.optimizers.SGD(lr=0.01)

 

# cost/loss 함수로 컴파일 작업
model.compile(loss='mean_squared_error', optimizer=sgd)

 

# 훈련 (1000번 반복)
model.fit(x_train, y_train, epochs=100)  # 100번 반복 훈련

 

# 테스트 모델
print(model.predict(np.array([5])))  # 입력 5에 대한 예측 값 출력

 

[결과 화면]

Train on 4 samples
Epoch 1/100
4/4 [==============================] - 1s 139ms/sample - loss: 1.3201
Epoch 2/100
4/4 [==============================] - 0s 4ms/sample - loss: 1.2937
Epoch 3/100
4/4 [==============================] - 0s 5ms/sample - loss: 1.2748
Epoch 4/100
4/4 [==============================] - 0s 5ms/sample - loss: 1.2592
...

...

 

 

<이상>