numerai_NN_example
Numerai tournament example scripts using pytorch NN, lightGBM and optuna
Performance of my model based on this example
Features
- era-boosted train, time-series cross-validation
- era-batches training
- model hyperparameter tuning on pytorch NN and GBDT model
- several tips on Numerai Forum are also included
Prerequisites
python3
gpu environment for pytorch # if you use pytorch NN model
virtualenv
Get this code and build environment
git clone https://github.com/meaten/numerai_NN_example.git
cd numerai_NN_example
mkdir env
virtualenv env -p python3
source env/bin/activate
pip install -r requirements.txt
Quick demo
train model by era-boosted training. you can choose other config files also.
python src/main.py --config_file config/mlp.yml --gpu GPU_ID
test model for diagnostic.
python src/main.py --config_file config/mlp.yml --mode test --gpu GPU_ID
inference & submit.
Please specify follows.
- pairs of your model name and config file in src/main.py.
- Numerai API user id and secret key in src/default_param.py
python src/main.py --mode submit --gpu GPU_ID
tune model hyperparameter by optuna To train with tuned parameters, add LOAD_TUNED: True
to the config file.
python src/main.py --config_file config/mlp.yml --mode tune --gpu GPU_ID
LICENCE
MIT
FEEDBACK
Please send me the bug report or wanted features on GitHub Issue or Numerai Forum.
SUPPORT
If you find this repository helpful and feel generous, Please send NMR to my wallet address below.
0x0000000000000000000000000000000000025769