Python mdnrnn_gym示例

示例#1

文件： state_embed_gym.py 项目： nbcmguarin0/ReAgent

def create_mdnrnn_trainer_and_embed_dataset(mdnrnn_params: OpenAiGymParameters,
                                            use_gpu):
    env, mdnrnn_trainer, _, _, _ = mdnrnn_gym(mdnrnn_params)
    embed_rl_dataset = RLDataset("/tmp/rl.pkl")
    create_embed_rl_dataset(env, mdnrnn_trainer, embed_rl_dataset, use_gpu,
                            mdnrnn_params.run_details)
    return env, mdnrnn_trainer, embed_rl_dataset

示例#2

文件： test_mdnrnn_gym.py 项目： lanfeng4659/Horizon

 def _test_mdnrnn(self,
                  params,
                  use_gpu=False,
                  feature_importance=False,
                  feature_sensitivity=False):
     return mdnrnn_gym(params, use_gpu, feature_importance,
                       feature_sensitivity)

示例#3

文件： state_embed_gym.py 项目： FlorianASchroeder/Horizon

def create_mdnrnn_trainer_and_embed_dataset(mdnrnn_params, use_gpu):
    env, mdnrnn_trainer, _, _, _ = mdnrnn_gym(mdnrnn_params, use_gpu)
    embed_rl_dataset = RLDataset("/tmp/rl.pkl")
    create_embed_rl_dataset(
        env, mdnrnn_trainer, embed_rl_dataset, use_gpu, **mdnrnn_params["run_details"]
    )
    return env, mdnrnn_trainer, embed_rl_dataset

示例#4

文件： test_mdnrnn_gym.py 项目： brettkoonce/ReAgent

 def _test_mdnrnn(
     self,
     params: OpenAiGymParameters,
     use_gpu=False,
     feature_importance=False,
     feature_sensitivity=False,
 ):
     return mdnrnn_gym(params, feature_importance, feature_sensitivity)