FastAPI ML Template

導入ライブラリー

• black: コードフォーマッター
• isort: import 整理ツール
• flake8: 品質チェック
• mypy: 型チェック
• pre-commit:コミット前に行うチェックを管理するツール
• pipenv: 仮想環境,パッケージ管理

Run Web API

Local

$ sh run.sh

Docker

$ docker build -f Dockerfile -t fastapi-ml .
$ docker run -p 9000:9000 --rm --name fastapi-ml -t -i fastapi-ml

Docker Compose

$ docker compose up --build

ディレクトリ構成

app
├── api
│   ├── api.py           # 機械学習の推論のエンドポイント
│   └── heartbeat.py     # WebAPIの外部監視用のエンドポイント
├── core
│   ├── config.py        # WebAPI全体の設定
│   └── event_handler.py # WebAPI起動/終了時に実行する処理
├── main.py              # WebAPI本体
├── models
│   └── predict.py       # WebAPIの入力/出力のモデル
└── services
    └── model.py         # 機械学習の推論の実装

参考

• 機械学習の推論 WebAPI の実装をテンプレート化して使い回せるようした