revive.algo package

Subpackages

• revive.algo.policy package
  • Submodules
  • revive.algo.policy.base module
    • catch_error()
    • PolicyOperator
      • PolicyOperator.env
      • PolicyOperator.train_policy
      • PolicyOperator.val_policy
      • PolicyOperator.policy
      • PolicyOperator.other_train_models
      • PolicyOperator.other_val_models
      • PolicyOperator.PARAMETER_DESCRIPTION
      • PolicyOperator.get_parameters()
      • PolicyOperator.get_tune_parameters()
      • PolicyOperator.optimizer_creator()
      • PolicyOperator.data_creator()
      • PolicyOperator.model_creator()
      • PolicyOperator.bc_optimizer_creator()
      • PolicyOperator.get_ope_dataset()
      • PolicyOperator.venv_test()
      • PolicyOperator.generate_rollout()
      • PolicyOperator.before_train_epoch()
      • PolicyOperator.after_train_epoch()
      • PolicyOperator.before_validate_epoch()
      • PolicyOperator.after_validate_epoch()
      • PolicyOperator.train_epoch()
      • PolicyOperator.bc_train_batch()
      • PolicyOperator.validate()
      • PolicyOperator.train_batch()
      • PolicyOperator.validate_batch()
    • PolicyAlgorithm
      • PolicyAlgorithm.get_train_func()
      • PolicyAlgorithm.get_trainer()
      • PolicyAlgorithm.get_trainable()
      • PolicyAlgorithm.get_parameters()
      • PolicyAlgorithm.get_tune_parameters()
  • revive.algo.policy.ppo module
    • PPOOperator
      • PPOOperator.NAME
      • PPOOperator.PARAMETER_DESCRIPTION
      • PPOOperator.policy_bc_optimizer
      • PPOOperator.model_creator()
      • PPOOperator.optimizer_creator()
      • PPOOperator.data_creator()
      • PPOOperator.after_train_epoch()
      • PPOOperator.train_batch()
      • PPOOperator.ADV()
      • PPOOperator.ppo_step()
  • revive.algo.policy.sac module
    • ReplayBuffer
      • ReplayBuffer.put()
      • ReplayBuffer.__len__()
      • ReplayBuffer.sample()
    • SACOperator
      • SACOperator.NAME
      • SACOperator.critic_pre_trained
      • SACOperator.PARAMETER_DESCRIPTION
      • SACOperator.model_creator()
      • SACOperator.optimizer_creator()
      • SACOperator.data_creator()
      • SACOperator.setup()
      • SACOperator.before_validate_epoch()
      • SACOperator.done_process()
      • SACOperator.buffer_process()
      • SACOperator.pretrain_critic()
      • SACOperator.bc_train_batch()
      • SACOperator.train_batch()
      • SACOperator.compute_lcb()
      • SACOperator.sac()
  • Module contents
• revive.algo.venv package
  • Submodules
  • revive.algo.venv.base module
    • catch_error()
    • VenvOperator
      • VenvOperator.NAME
      • VenvOperator.metric_name
      • VenvOperator.nodes_models_train
      • VenvOperator.other_models_train
      • VenvOperator.nodes_models_val
      • VenvOperator.other_models_val
      • VenvOperator.PARAMETER_DESCRIPTION
      • VenvOperator.get_parameters()
      • VenvOperator.get_tune_parameters()
      • VenvOperator.model_creator()
      • VenvOperator.optimizer_creator()
      • VenvOperator.data_creator()
      • VenvOperator.nan_in_grad()
      • VenvOperator.before_train_epoch()
      • VenvOperator.after_train_epoch()
      • VenvOperator.before_validate_epoch()
      • VenvOperator.after_validate_epoch()
      • VenvOperator.train_epoch()
      • VenvOperator.validate_epoch()
      • VenvOperator.train_batch()
      • VenvOperator.validate_batch()
    • VenvAlgorithm
      • VenvAlgorithm.get_train_func()
      • VenvAlgorithm.get_trainer()
      • VenvAlgorithm.get_trainable()
      • VenvAlgorithm.get_parameters()
      • VenvAlgorithm.get_tune_parameters()
  • revive.algo.venv.bc module
    • BCOperator
      • BCOperator.NAME
      • BCOperator.PARAMETER_DESCRIPTION
      • BCOperator.model_creator()
      • BCOperator.optimizer_creator()
      • BCOperator.data_creator()
      • BCOperator.train_epoch()
      • BCOperator.train_batch()
  • revive.algo.venv.revive module
    • ReviveOperator
      • ReviveOperator.NAME
      • ReviveOperator.matcher_model_creator()
      • ReviveOperator.model_creator()
      • ReviveOperator.data_creator()
      • ReviveOperator.switch_mail_data_loader()
      • ReviveOperator.bc_train_batch()
      • ReviveOperator.train_epoch()
      • ReviveOperator.mix_data_process()
      • ReviveOperator.net_l2_norm()
  • revive.algo.venv.revive_f module
    • FILTEROperator
      • FILTEROperator.NAME
      • FILTEROperator.PARAMETER_DESCRIPTION
      • FILTEROperator.nodes_models_mail
      • FILTEROperator.other_models_mail
      • FILTEROperator.data_creator()
      • FILTEROperator.mail_model_creator()
      • FILTEROperator.bc_model_creator()
      • FILTEROperator.matcher_model_creator()
      • FILTEROperator.mail_generator_model_creator()
      • FILTEROperator.bc_optimizer_creator()
      • FILTEROperator.mail_optimizer_creator()
      • FILTEROperator.train_epoch()
      • FILTEROperator.PPO_step()
  • revive.algo.venv.revive_p module
    • PPOOperator
      • PPOOperator.NAME
      • PPOOperator.PARAMETER_DESCRIPTION
      • PPOOperator.generator_model_creator()
      • PPOOperator.optimizer_creator()
      • PPOOperator.ADV()
      • PPOOperator.PPO_step()
  • revive.algo.venv.revive_t module
    • ReplayBuffer
      • ReplayBuffer.put()
      • ReplayBuffer.__len__()
      • ReplayBuffer.sample()
    • TD3Operator
      • TD3Operator.NAME
      • TD3Operator.PARAMETER_DESCRIPTION
      • TD3Operator.setup()
      • TD3Operator.generator_model_creator()
      • TD3Operator.optimizer_creator()
      • TD3Operator.TD3_step()
  • revive.algo.venv.template module
    • AlgorithmOperator
      • AlgorithmOperator.NAME
      • AlgorithmOperator.PARAMETER_DESCRIPTION
      • AlgorithmOperator.get_parameters()
      • AlgorithmOperator.get_tune_parameters()
      • AlgorithmOperator.model_creator()
      • AlgorithmOperator.optimizer_creator()
      • AlgorithmOperator.data_creator()
      • AlgorithmOperator.train_epoch()
      • AlgorithmOperator.train_batch()
  • Module contents

Module contents