Skip to content

Commit

Permalink
Create README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
jspark418 authored Apr 15, 2023
1 parent ea189a3 commit 82440cc
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
# Energy Forecasting 모델
## 프로젝트 소개
* 다종(전기,수도,가스,온수,열)의 에너지 사용량을 입력 받아 특정 에너지의 사용량을 예측하는 모델입니다.
* 해당 모델은 multi-modal learning 구조를 가지며 개별 에너지원은 하나의 모달리티를 의미합니다.
* 각각의 에너지 사용량은 feature extractor를 거쳐 예측하고자 하는 에너지와 관련된 feature vector를 추출합니다.
* 추출된 각 에너지의 feature vector는 concatenation 된 뒤 Fully-connected layer를 통과해 원하는 에너지의 사용량을 예측합니다.

## Setup
### Requirements
* python 3.7.12
* tensorflow 2.6.4
### Docker 환경 setup
* 본 프로젝트는 Ubuntu 18.04.6 LTS, docker 환경에서 진행되었습니다.
* docker image 설치
```
sudo docker pull gcr.io/kaggle-gpu-images/python:v121
```

## Models
* 현재 feature extractor에 사용된 모델은 transformer 입니다.
* transformer 외에 다른 모델을 사용하고 싶으면 model.py 에 작성된 모델의 코드를 수정하시면 됩니다.

0 comments on commit 82440cc

Please sign in to comment.