Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Danyeong-Lee committed Jun 11, 2023
1 parent e00a289 commit 05ac866
Showing 1 changed file with 19 additions and 1 deletion.
20 changes: 19 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,4 +12,22 @@ ______________________________________________________________________
구정현, 이단영, 김상엽

## Solution Overview
(WIP)
(wip)


## Running
### Download dataset
- Training and test data can be downloaded from [2022 Samsung AI Challenge (Materials Discovery)](https://dacon.io/competitions/official/235953/data).
- We assume that you appropriately downloaded the dataset into 'data' directory.

### Train single model
```
python train.py configs/gem1.yaml
```
- You can train your model using default hyperparameters we used in competition with above command.
- Trained model checkpoints and submission files (test_preds.csv) will be saved in 'outputs' directory. You can directly submit the csv file.
- Just a single model achieved high performance (public score: 0.65 ~ 0.7), but for further improvement, we used stacking ensemble.

### Stacking ensemble
(wip)

0 comments on commit 05ac866

Please sign in to comment.