PyTorchで強化学習アルゴリズムを実装したライブラリ。研究や実用で信頼でき、一貫性のあるベースラインを提供します。
DLR-RM/rl-baselines3-zoo
Stable Baselines3の強化学習エージェント向けトレーニングフレームワーク。調整済みハイパーパラメータ、学習済みエージェント、ハイパーパラメータ最適化スクリプトを提供します。
- スター
- 2,900
- フォーク
- 612
- 未解決の Issue
- 62
- 最終コミット
- 5 日前
これは何?
RL Baselines3 Zooは、Stable Baselines3を基盤とする強化学習向けのトレーニングフレームワークです。エージェントの学習と評価、ハイパーパラメータの調整、結果のプロット、エージェントの動画撮影に使えるCLIスクリプトを提供します。
一般的な環境やアルゴリズム向けに調整済みのハイパーパラメータと、それらの設定で学習したエージェントも収録しています。ハイパーパラメータは環境ごとにYAMLファイルで定義されているため、アルゴリズムと環境IDを指定するだけで学習を開始できます。学習済みエージェントはenjoyスクリプトで再生でき、対応環境にはAtari、classic control、Box2D、PyBullet、MuJoCo、robotics、MiniGridが含まれます。
実験のトラッキングにはWeights & Biases、学習済みモデルの保存や共有にはHugging Faceとの連携を利用できます。強化学習エージェントの学習、アルゴリズムのベンチマーク、学習済み設定の再利用を手軽に行いたいユーザーを対象としています。
主な機能
- 01エージェントの学習、評価、再生に使えるCLIスクリプト
- 02環境とアルゴリズムごとに調整された、YAMLファイルで定義されるハイパーパラメータ
- 03ハイパーパラメータ最適化ツール
- 04Hugging Faceで利用できる学習済みエージェント
- 05Weights & BiasesおよびHugging Faceとの連携
- 06学習結果をプロットするスクリプト
ユースケース
- 定型コードを書かずにGymやPyBulletの環境で強化学習エージェントを学習する
- 同じタスクで異なる強化学習アルゴリズムをベンチマークする
- 新しい環境に適したハイパーパラメータを探す
- 学習済みエージェントを再利用して、ポリシーを確認・評価する
クイックスタート
- 1
PyPI から rl_zoo3 パッケージをインストールします
pip install rl_zoo3 - 2
クローンしたリポジトリから、定期的な評価を行いながら HalfCheetahBulletEnv-v0 で SAC エージェントを学習させます
python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1
使う前に知っておきたいこと
- 完全なインストールには、swig、cmake、ffmpeg などのシステムパッケージが必要です。
- README では、特別な CFLAGS の回避策を使わないと macOS で pybullet のビルドに失敗する可能性があると警告しています。
- ベンチマーク結果は各1回の実行に基づくため、定量的なベンチマークではありません。
トピック
よくある質問
- rl-baselines3-zoo とは?
- Stable Baselines3の強化学習エージェント向けトレーニングフレームワーク。調整済みハイパーパラメータ、学習済みエージェント、ハイパーパラメータ最適化スクリプトを提供します。 RL Baselines3 Zooは、Stable Baselines3を基盤とする強化学習向けのトレーニングフレームワークです。エージェントの学習と評価、ハイパーパラメータの調整、結果のプロット、エージェントの動画撮影に使えるCLIスクリプトを提供します。
- rl-baselines3-zoo は無料で使えますか?ライセンスは?
- rl-baselines3-zoo は MIT ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- rl-baselines3-zoo は今もメンテナンスされていますか?
- 活発にメンテナンス中。rl-baselines3-zoo の最終コミットは 2026年10月16日 です。
- rl-baselines3-zoo は何に使えますか?
- 定型コードを書かずにGymやPyBulletの環境で強化学習エージェントを学習する 同じタスクで異なる強化学習アルゴリズムをベンチマークする 新しい環境に適したハイパーパラメータを探す 学習済みエージェントを再利用して、ポリシーを確認・評価する
- rl-baselines3-zoo のインストール方法は?
- PyPI から rl_zoo3 パッケージをインストールします: pip install rl_zoo3 クローンしたリポジトリから、定期的な評価を行いながら HalfCheetahBulletEnv-v0 で SAC エージェントを学習させます: python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1
- rl-baselines3-zoo はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/DLR-RM/rl-baselines3-zoo
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
自動株式取引やポートフォリオ運用に深層強化学習を適用するための、オープンソースのPythonフレームワークです。
Physical Intelligenceによるオープンソースのロボット向けVLAモデル。事前学習済みチェックポイントとファインチューニングの例を提供します。
単一エージェントの強化学習環境向け標準Python APIと、OpenAI Gymからフォークした参照環境群。
Mambaは、選択的状態空間モデルを基盤とするPyTorchのシーケンスモデリングアーキテクチャで、言語などのシーケンス処理においてTransformerに代わる線形時間の手法を提供します。
単一の24GB GPUで実行できる、小規模な中国語LLaMA2系モデル向けの事前学習・SFTパイプラインをゼロから構築。