将棋の次の一手予測モデルと、その推論結果を確認するための Web アプリです。
web/: React + Vite のフロントエンドserver/: FastAPI の推論 APIshogi_ai/: 学習コード、前処理、モデル定義、学習成果物
このプロジェクトは Kaggle の Elite Shogi Games を学習データ取得元として想定しています。
- Dataset:
Elite Shogi Games - Author:
suayptalha - URL:
https://www.kaggle.com/datasets/suayptalha/elite-shogi-games
方針:
shogi_games.csvはリポジトリに含めません- 必要な場合は Kaggle から各自ダウンロードしてください
- 学習済みモデル
shogi_ai/best_model.ptはリポジトリに同梱します
cd server
python3 -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
uvicorn app:app --host 0.0.0.0 --port 8000 --reloadGET /health で疎通確認できます。
cd web
cp .env.example .env.local
npm install
npm run devVITE_API_BASE_URL を変えると、接続先 API を切り替えられます。未指定時は http://localhost:8000 を使います。
server/はshogi_ai/のmodel.py、shogi_env.py、best_model.ptを参照しますshogi_ai/dataset.npzは再生成できる中間生成物として.gitignoreに入れていますshogi_ai/shogi_games.csvは Kaggle 取得前提なので.gitignoreに入れています
- このリポジトリは MIT License です
- 同梱している
shogi_ai/best_model.ptも、特に別記がない限り同じ MIT License で扱う想定です - 学習元データ
shogi_games.csvは第三者データなので、このリポジトリから再配布しない方針です