from tensorflow.python.keras.models import Sequential
from tensorflow.python.keras.layers import Dense
import numpy as np
import matplotlib.pyplot as plt

#1. 데이터
x_train = np.array(range(1,11)) # 스칼라 10개가 모인 벡터 1개
y_train = np.array(range(1,11))

x_val = np.array([14,15,16])    #  13개의 데이터중에 3개는 발리데이션, 3개는 테스트, 나머지는 트레인
y_val = np.array([14,15,16])

x_test = np.array([11,12,13])
y_test = np.array([11,12,13])

#2. 모델구성
model = Sequential()
model.add(Dense(5, activation='linear', input_dim=1))
model.add(Dense(10, activation='relu'))
model.add(Dense(15, activation='relu'))
model.add(Dense(20, activation='relu'))
model.add(Dense(30, activation='relu'))
model.add(Dense(40, activation='relu'))
model.add(Dense(30, activation='relu'))
model.add(Dense(25, activation='relu'))
model.add(Dense(20, activation='relu'))
model.add(Dense(10, activation='relu'))
model.add(Dense(5, activation='relu'))
model.add(Dense(1))

#3. 컴파일, 훈련
model.compile(loss='mse', optimizer='adam')
model.fit(x_train, y_train, epochs=200, batch_size=1, verbose=1,
          validation_data=(x_val, y_val)
          )

#4. 평가, 예측
loss = model.evaluate(x_test, y_test)
print('loss :', loss)

result = model.predict([17])
print('17의 예측값 :', result)

#[[16.925383]]

#loss : 2.3002474335953593e-05
# 17의 예측값 : [[17.001276]]

# loss : 1.2247861613801092e-09
# 17의 예측값 : [[16.999826]]