더북(TheBook)

이제 인공지능 모델을 개발하기 위한 데이터를 생성하겠습니다.

look_back = 3

며칠 동안의 연속된 데이터를 바탕으로 인공지능에서 학습할지를 정해주는 코드입니다. 이번 예제에서는 그 수를 3으로 설정(look_back = 3)하겠습니다(숫자를 변경해도 됩니다).

x_train, y_train = create_dataset(train_data, look_back)

훈련 데이터를 생성하기 위해 17.4절에서 만든 create_dataset 함수를 호출하겠습니다. 함수에 전달할 인자 중 첫 번째 인자에 훈련 데이터(train_data)를 두 번째 인자에 look_back(3을 담고 있습니다)을 넣습니다.

함수를 호출한 결과 2개의 데이터가 생성(x_data, y_data)됩니다. 이제 이 함수를 호출하면 두 개의 배열을 반환해 줍니다. 첫 번째는 3일치의 연속된 값(x_data)으로 이루어진 데이터를, 두 번째는 인공지능이 학습할 값(y_data)입니다.

첫 번째 생성되는 x_datax_train 데이터로, 두 번째 생성되는 y_datay_train 데이터로 넣습니다.

x_test, y_test = create_dataset(test_data, look_back)

검증 데이터를 생성하기 위해 create_dataset 함수를 호출하겠습니다. 함수에 전달할 인자 중 첫 번째 인자에 검증 데이터(test_data)를 넣습니다. 그리고 두 번째 인자에 look_back을 넣습니다. 마찬가지로 첫 번째 생성되는 x_datax_test 데이터로, 두 번째 생성되는 y_datay_test 데이터로 넣습니다.

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