移動

• 説明:

データセットは、MuJoCo 移動タスクの環境報酬でトレーニングされた SAC エージェントを使用して作成されました。これらのデータセットは、敵対的模倣学習の重要事項で使用されていますか?オルシーニ等。 2021 .

データセットは、ステップとエピソードを表すRLDS 形式に従います。

• ホームページ: https://github.com/google-research/rlds

• ソース コード: tfds.rlds.locomotion.Locomotion

• バージョン:

  • 1.0.0 (デフォルト): 初期リリース。

• 監視されたキー( as_supervised docを参照): None

• 図( tfds.show_examples ): サポートされていません。

• 引用：

@article{orsini2021matters,
  title={What Matters for Adversarial Imitation Learning?},
  author={Orsini, Manu and Raichuk, Anton and Hussenot, L{'e}onard and Vincent, Damien and Dadashi, Robert and Girgin, Sertan and Geist, Matthieu and Bachem, Olivier and Pietquin, Olivier and Andrychowicz, Marcin},
  journal={International Conference in Machine Learning},
  year={2021}
}

locomotion/ant_sac_1M_single_policy_stochastic (デフォルト設定)

• 構成の説明: Ant 用に 1M ステップのトレーニングを受けた SAC エージェントによって生成されたデータセット。

• ダウンロードサイズ: 6.49 MiB

• データセットサイズ: 23.02 MiB

• 自動キャッシュ(ドキュメント): はい

• スプリット:

• 機能構造:

FeaturesDict({
    'steps': Dataset({
        'action': Tensor(shape=(8,), dtype=float32),
        'discount': float32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': Tensor(shape=(111,), dtype=float32),
        'reward': float32,
    }),
})

• 機能のドキュメント:

• 例( tfds.as_dataframe ):

locomotion/hopper_sac_1M_single_policy_stochastic

• 構成の説明: ホッパーの 1M ステップのトレーニングを受けた SAC エージェントによって生成されたデータセット。

• ダウンロードサイズ: 2.26 MiB

• データセットサイズ: 2.62 MiB

• 自動キャッシュ(ドキュメント): はい

• スプリット:

• 機能構造:

FeaturesDict({
    'steps': Dataset({
        'action': Tensor(shape=(3,), dtype=float32),
        'discount': float32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': Tensor(shape=(11,), dtype=float32),
        'reward': float32,
    }),
})

• 機能のドキュメント:

• 例( tfds.as_dataframe ):

ロコモーション/halfcheetah_sac_1M_single_policy_stochastic

• 構成の説明: HalfCheetah の 1M ステップのトレーニングを受けた SAC エージェントによって生成されたデータセット。

• ダウンロードサイズ: 4.49 MiB

• データセットのサイズ: 4.93 MiB

• 自動キャッシュ(ドキュメント): はい

• スプリット:

• 機能構造:

FeaturesDict({
    'steps': Dataset({
        'action': Tensor(shape=(6,), dtype=float32),
        'discount': float32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': Tensor(shape=(17,), dtype=float32),
        'reward': float32,
    }),
})

• 機能のドキュメント:

• 例( tfds.as_dataframe ):

locomotion/walker2d_sac_1M_single_policy_stochastic

• 構成の説明: Walker2d の 1M ステップのトレーニングを受けた SAC エージェントによって生成されたデータセット。

• ダウンロードサイズ: 4.35 MiB

• データセットのサイズ: 4.91 MiB

• 自動キャッシュ(ドキュメント): はい

• スプリット:

• 機能構造:

FeaturesDict({
    'steps': Dataset({
        'action': Tensor(shape=(6,), dtype=float32),
        'discount': float32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': Tensor(shape=(17,), dtype=float32),
        'reward': float32,
    }),
})

• 機能のドキュメント:

• 例( tfds.as_dataframe ):

locomotion/humanoid_sac_15M_single_policy_stochastic

• 構成の説明: ヒューマノイド用に 1,500 万ステップのトレーニングを受けた SAC エージェントによって生成されたデータセット。

• ダウンロードサイズ: 192.78 MiB

• データセットのサイズ: 300.94 MiB

• 自動キャッシュ(ドキュメント): いいえ

• スプリット:

• 機能構造:

FeaturesDict({
    'steps': Dataset({
        'action': Tensor(shape=(17,), dtype=float32),
        'discount': float32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': Tensor(shape=(376,), dtype=float32),
        'reward': float32,
    }),
})

• 機能のドキュメント:

• 例( tfds.as_dataframe ):