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')
    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.