더북(TheBook)

 

2선형 회귀 실행

 

선형 회귀 데이터는 마지막에 참과 거짓을 구분할 필요가 없습니다. 출력층에 활성화 함수를 지정할 필요도 없습니다.

 

model = Sequential()

model.add(Dense(30, input_dim=13, activation=‘relu’))

model.add(Dense(6, activation=‘relu’))

model.add(Dense(1))

 

모델의 학습이 어느 정도 되었는지 확인하기 위해 예측 값과 실제 값을 비교하는 부분을 추가합니다.

 

Y_prediction = model.predict(X_test).flatten()

for i in range(10):

    label = Y_test[i]

    prediction = Y_prediction[i]

    print(“실제가격: {:.3f}, 예상가격: {:.3f}”.format(label, prediction))

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.