Ejemplos de mdnrnn_gym en Python

Ejemplo n.º 1

Archivo: state_embed_gym.py Proyecto: 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

Ejemplo n.º 2

Archivo: test_mdnrnn_gym.py Proyecto: 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)

Ejemplo n.º 3

Archivo: state_embed_gym.py Proyecto: 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

Ejemplo n.º 4

Archivo: test_mdnrnn_gym.py Proyecto: 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)