7.5.3 LSTM 계층 구현
LSTM 계층에 대한 예제 데이터셋은 스타벅스 주가 데이터셋을 사용합니다. 또한, 예제를 통해 확인하고자 하는 것은 모델이 얼마나 주가 예측을 잘 하는지에 대해 알아보는 것입니다.
먼저 필요한 라이브러리를 호출합니다.
코드 7-40 라이브러리 호출
import os
import time
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import torch
import torch.nn as nn
from torch.utils.data import TensorDataset, DataLoader
from torch.autograd import Variable
from tqdm import tqdm_notebook
from sklearn.preprocessing import StandardScaler, MinMaxScaler
from sklearn.model_selection import train_test_split
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')