Skip to content

⚙️ [기능추가][AI] Kaggle 기반 자동 학습 파이프라인 구축 #343

@vmgfh878-art

Description

@vmgfh878-art

📝 현재 문제점
운영 서버 메모리가 2GB로 제한되어 딥러닝 모델 학습이 불가능함.
PatchTST를 로컬 CPU로 실행 시 epoch 1개도 완료되지 않음.

🛠️ 해결 방안 / 제안 기능
학습은 Kaggle GPU, 추론은 서버로 분리하는 파이프라인 구축.

추가할 파일

  • AI/scripts/extract_to_parquet.py (DB → parquet 추출)
  • AI/scripts/upload_to_kaggle.py (Kaggle 데이터셋 업로드)
  • AI/scripts/trigger_training.py (노트북 학습 트리거)
  • AI/scripts/download_weights.py (가중치 다운로드)
  • AI/scripts/deploy_to_server.py (서버 배포)
  • AI/pipelines/weekly_routine.py (파이프라인 메인)
  • .github/workflows/train.yml (GitHub Actions 자동화)
  • AI/kaggle_notebooks/ (4개 모델 노트북)

수정할 파일

  • AI/modules/signal/models/PatchTST/wrapper.py
  • AI/modules/signal/models/TCN/wrapper.py

🙋‍♂️ 담당자

  • AI : 김지형

Metadata

Metadata

Assignees

Labels

AIAI 이슈

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions