Ayniy, All You Need is YAML
Ayniy is a supporting tool for machine learning competitions.
Documentation | Slide (Japanese)
# Import packages
from sklearn.model_selection import StratifiedKFold
import yaml
from ayniy.model.runner import Runner
# Load configs
f = open('configs/run000.yml', 'r+')
configs = yaml.load(f, Loader=yaml.SafeLoader)
# Difine CV strategy as you like
cv = StratifiedKFold(n_splits=5, shuffle=True, random_state=7)
# Modeling
runner = Runner(configs, cv)
runner.run_train_cv()
runner.run_predict_cv()
runner.submission()
Examples
Competition Name | Score (Rank) | Repository |
---|---|---|
Titanic: Machine Learning from Disaster | 0.76555 (-) | Link |
Basketball Behavior Challenge BBC2020 | 0.871700 (1) | Link |
ひろしまQuest2020#stayhome【アイデア部門】 | - | Link |
Environment
docker-compose -d --build
docker exec -it ayniy-test bash
MLflow
cd experiments
mlflow ui
Test
# pytest
pytest tests/ --cov=. --cov-report=html
# black
black . --line-length 99 --exclude="docs"
# flake8
flake8 .
# mypy
mypy .
Docs
In container,
cd docs
make html
Out of container,
sh deploy.sh