3.9 케라스로 모델 만들기
이제 케라스로 모델 아키텍처를 구성하자. 앞서 살펴본 2장과 마찬가지로 케라스의 Sequential 클래스를 사용해 레이어를 하나씩 쌓아 모델을 만든다.
먼저 DataFrame을 특징 변수(X)와 예측할 목표 변수(y)로 나눈다.
X = df.loc[:, df.columns != 'fare_amount'] y = df.loc[:, 'fare_amount']
그런 다음 데이터를 훈련 데이터셋(80%)과 테스트 데이터셋(20%)으로 나눈다.
from sklearn.model_selection import train_test_split X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)