이제 인공지능 모델을 개발하기 위한 데이터를 생성하겠습니다.
■ 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_data를 x_train 데이터로, 두 번째 생성되는 y_data를 y_train 데이터로 넣습니다.
■ x_test, y_test = create_dataset(test_data, look_back)
검증 데이터를 생성하기 위해 create_dataset 함수를 호출하겠습니다. 함수에 전달할 인자 중 첫 번째 인자에 검증 데이터(test_data)를 넣습니다. 그리고 두 번째 인자에 look_back을 넣습니다. 마찬가지로 첫 번째 생성되는 x_data를 x_test 데이터로, 두 번째 생성되는 y_data를 y_test 데이터로 넣습니다.