inflearn logo
강의

Course

Instructor

The Secret of Algorithmic Trading: How AI Predicts Stock Prices

Transformer Model Development Completed and Code Review

섹션4의 10강 질문이 있습니다.

69

gnb1202

1 asked

1

train_size = int(len(data) * 0.8) 
train_data = data.iloc[:train_size]       test_data = data.iloc[train_size:]       

훈련과 테스트 8:2로 잘 나누고

for i in range(lookback, len(data_scaled) - forecast_horizon):

이렇게 돌리면 그냥 전체 데이터를 학습한거 아닌가요?

결국 안나누고 학습한 것 같은데, 그럼 이 프로젝트는 그냥 데이터를 모두 학습한거죠?

LR 버전으로 리팩토링해서 돌려보고 있는데 과적합뜨는 것 보니까 그냥 완전 똑같이 나오더라구요.

python 딥러닝 재테크 lstm transformer

Answer 2

0

cheatkeylab

안녕하세요, @gnb1202님.

문제를 정확하게 잘 짚어주셨습니다.

현재 코드대로면 테스트 데이터까지 학습에 포함되어 있어서 lookahead bias가 발생합니다.

for i in range(lookback, len(data_scaled) - forecast_horizon):

위 for loop에 data_scaled전체 데이터(train + test)를 scaling한 것이기 때문.

우선 아래 두 군데를 변경해서 테스트해보시길 바랍니다.

저도 곧 업데이트 본 반영하도록 하겠습니다.

 

  1. 스케일러 부분

print("Scaling data...")
train_size = int(len(data) * 0.8)
train_data = data.iloc[:train_size]
test_data = data.iloc[train_size:]

# 스케일러는 train에만 fit !!!
stock_scaler = MinMaxScaler()
econ_scaler = MinMaxScaler()

# train 데이터로만 fit
stock_scaler.fit(train_data[target_columns])  
econ_scaler.fit(train_data[economic_features])

# 전체 데이터는 transform만
data_scaled = data.copy()
data_scaled[target_columns]    = stock_scaler.transform(data[target_columns])     # ← fit_transform → transform으로 변경
data_scaled[economic_features] = econ_scaler.transform(data[economic_features])    # ← fit_transform → transform으로 변경
  1. 학습 데이터 생성 부분

# 학습 데이터 생성 (데이터 누수 차단)
lookback = 90
forecast_horizon = 14

# train 구간 끝에서 forecast_horizon 만큼 빼야 y가 test 구간으로 안 넘어감
last_train_index = train_size - forecast_horizon  

X_stock_train = []
X_econ_train = []
y_train = []

for i in range(lookback, last_train_index):
    X_stock_train.append(data_scaled[target_columns].iloc[i-lookback:i].values)
    X_econ_train.append(data_scaled[economic_features].iloc[i-lookback:i].values)
    y_train.append(data_scaled[target_columns].iloc[i + forecast_horizon - 1].values)  # i+13은 아직 train 안에 있음

X_stock_train = np.array(X_stock_train = np.array(X_stock_train)
X_econ_train = np.array(X_econ_train)
y_train = np.array(y_train)

작업 하시다가 잘 안되는 부분 있으시다면 다시 말씀 주세요~

감사합니다.

0

communityai8509

안녕하세요, 인프런 AI 인턴입니다.
gnb1202님, 데이터셋을 훈련과 테스트로 8:2로 나누신 방식은 정확히 사용하신 것 같습니다.
두 개의 데이터프레임, train_datatest_data를 만드셨는데요,

train_size = int(len(data) * 0.8)
train_data = data.iloc[:train_size]
test_data = data.iloc[train_size:]

나누신 이후에, 전체 데이터를 학습 데이터와 비슷하게 학습하고 싶지 않다면, for 루프의 데이터 사용 범위를 조정하셔야 합니다.

for 루프에서 lookbackforecast_horizon을 사용한 코드는 특정 범위를 제외하고 데이터를 사용하는 것으로 보입니다. 하지만, 이 부분이 전체 데이터에서 사용되었므로 훈련 데이터 범위 내에서만 실행되도록 코드를 조정하셔야 합니다. 그렇지 않으면 전체 데이터를 학습하게 될 가능성이 있습니다.

과적합에 대한 문제는 학습 데이터와 테스트 데이터 간의 경계를 명확히 하고, lookbackforecast_horizon 변수를 조정하여 해결할 수 있습니다. 모델의 과적합을 방지하려면 모델의 복잡성을 낮추거나 정규화 등의 기법을 사용하는 것도 고려해보세요.

이미지를 참고하시려면 문의 게시판에서 직접 질문을 올리신 후, 공유 공유자님께 추가적인 설명을 받아보는 것도 좋은 방법입니다. 수강 전 문의 게시판에서 정보를 확인해 문의를 남기시면 보다 정확한 피드백을 받으실 수 있습니다.

36. 텔레그램 연동(특강) 노션 링크

0

6

1

수강기간연장 문의드립니다

0

7

1

수강 연장 문의

0

10

1

label 인코딩 에러 발생 원인

0

11

2

LabelEncoding 시 train data 학습(fit_transform, transform)

0

13

2

로지스틱 회귀분석은 옛날거 밖에 없나요?

0

22

2

강의 초반 질문드려요!

0

13

1

사주 정보를 넣고 웹훅

0

17

2

오류 문구 입니다.

0

20

1

creatomate를 대체할 만한 곳

0

19

1

Part2 - 19번 ~ 20번 강의 사이에 일부 내용이 생략된듯합니다.

0

23

2

Gemini 모델 변경 문의

0

21

2

소스코드 다운로드 어떻게 받는지 물어보려고 했는데...

0

12

1

판다스 표 표시

0

19

3

part2 gemini api 연동

0

20

3

알라딘 서점에서 도서목록 관련 Request정보를 찾을 수가 없습니다.

0

21

3

주가 예측 정확도

1

118

2

세션 3까지 듣고 궁금한점이 있어 문의 드립니다,

0

79

2

파이썬 스크립트 작성 프로그램은 뭘로 쓰시는건가요?

1

245

2

DB에 데이터 넣지 않는 이유가 무엇때문인가요?

1

111

1

국내시장용으로 제작하려면 어떻게 해야하나요?

0

134

2

feature 질문 드립니다.

0

76

2

노션 및 수업자료(코드) 다운로드는 어떻게 하나요?

1

199

2

주식 종목 추가

0

240

1