├── .gitignore ├── LICENSE ├── README.md ├── _assets └── gfn_maxent_rl.png ├── config ├── algorithm │ ├── fldb.yaml │ ├── mdb.yaml │ ├── pcl.yaml │ ├── pisql.yaml │ ├── sac.yaml │ ├── sql.yaml │ └── tb.yaml ├── default.yaml └── env │ ├── dag_gfn.yaml │ ├── phylo_gfn.yaml │ └── treesample.yaml ├── gfn_maxent_rl ├── __init__.py ├── algos │ ├── __init__.py │ ├── base.py │ ├── forward_looking_detailed_balance.py │ ├── modified_detailed_balance.py │ ├── path_consistency_learning.py │ ├── soft_actor_critic.py │ ├── soft_qlearning.py │ ├── soft_qlearning_policy.py │ └── trajectory_balance.py ├── data │ ├── __init__.py │ ├── replay_buffer.py │ └── replay_buffer_episodic.py ├── envs │ ├── __init__.py │ ├── dag_gfn │ │ ├── __init__.py │ │ ├── base.py │ │ ├── data_generation │ │ │ ├── __init__.py │ │ │ ├── create_artifacts.py │ │ │ ├── data.py │ │ │ └── graphs.py │ │ ├── env.py │ │ ├── factories.py │ │ ├── functional.py │ │ ├── graph_priors.py │ │ ├── jraph_utils.py │ │ ├── policy.py │ │ ├── scores.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── exhaustive.py │ │ │ └── graphs.py │ ├── errors.py │ ├── phylo_gfn │ │ ├── __init__.py │ │ ├── datasets │ │ │ ├── DS1.json │ │ │ ├── DS2.json │ │ │ ├── DS3.json │ │ │ ├── DS4.json │ │ │ ├── DS5.json │ │ │ ├── DS6.json │ │ │ ├── DS7.json │ │ │ └── DS8.json │ │ ├── env.py │ │ ├── factories.py │ │ ├── functional.py │ │ ├── policy.py │ │ ├── rewards.py │ │ ├── trees.py │ │ └── utils.py │ ├── treesample │ │ ├── __init__.py │ │ ├── chain_env.py │ │ ├── factor_graph1_env.py │ │ ├── factor_graph2_env.py │ │ ├── factor_graph_env.py │ │ ├── factories.py │ │ ├── functional.py │ │ ├── permuted_chain_env.py │ │ ├── policy.py │ │ └── wrappers.py │ └── wrappers.py ├── nets │ ├── __init__.py │ ├── gnn.py │ └── transformer.py └── utils │ ├── __init__.py │ ├── async_evaluation.py │ ├── beam_search.py │ ├── estimation.py │ ├── evaluations.py │ ├── exhaustive.py │ ├── io.py │ └── metrics.py ├── requirements.txt └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/README.md -------------------------------------------------------------------------------- /_assets/gfn_maxent_rl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/_assets/gfn_maxent_rl.png -------------------------------------------------------------------------------- /config/algorithm/fldb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/fldb.yaml -------------------------------------------------------------------------------- /config/algorithm/mdb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/mdb.yaml -------------------------------------------------------------------------------- /config/algorithm/pcl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/pcl.yaml -------------------------------------------------------------------------------- /config/algorithm/pisql.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/pisql.yaml -------------------------------------------------------------------------------- /config/algorithm/sac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/sac.yaml -------------------------------------------------------------------------------- /config/algorithm/sql.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/sql.yaml -------------------------------------------------------------------------------- /config/algorithm/tb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/algorithm/tb.yaml -------------------------------------------------------------------------------- /config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/default.yaml -------------------------------------------------------------------------------- /config/env/dag_gfn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/env/dag_gfn.yaml -------------------------------------------------------------------------------- /config/env/phylo_gfn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/env/phylo_gfn.yaml -------------------------------------------------------------------------------- /config/env/treesample.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/config/env/treesample.yaml -------------------------------------------------------------------------------- /gfn_maxent_rl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/base.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/forward_looking_detailed_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/forward_looking_detailed_balance.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/modified_detailed_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/modified_detailed_balance.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/path_consistency_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/path_consistency_learning.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/soft_actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/soft_actor_critic.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/soft_qlearning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/soft_qlearning.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/soft_qlearning_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/soft_qlearning_policy.py -------------------------------------------------------------------------------- /gfn_maxent_rl/algos/trajectory_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/algos/trajectory_balance.py -------------------------------------------------------------------------------- /gfn_maxent_rl/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/data/__init__.py -------------------------------------------------------------------------------- /gfn_maxent_rl/data/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/data/replay_buffer.py -------------------------------------------------------------------------------- /gfn_maxent_rl/data/replay_buffer_episodic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/data/replay_buffer_episodic.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/__init__.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/base.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/data_generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/data_generation/create_artifacts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/data_generation/create_artifacts.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/data_generation/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/data_generation/data.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/data_generation/graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/data_generation/graphs.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/factories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/factories.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/functional.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/graph_priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/graph_priors.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/jraph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/jraph_utils.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/policy.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/scores.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/utils/exhaustive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/utils/exhaustive.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/dag_gfn/utils/graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/dag_gfn/utils/graphs.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/errors.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/__init__.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS1.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS2.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS3.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS4.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS5.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS6.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS6.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS7.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS7.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/datasets/DS8.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/datasets/DS8.json -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/factories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/factories.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/functional.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/policy.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/rewards.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/trees.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/trees.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/phylo_gfn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/phylo_gfn/utils.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/__init__.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/chain_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/chain_env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/factor_graph1_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/factor_graph1_env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/factor_graph2_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/factor_graph2_env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/factor_graph_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/factor_graph_env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/factories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/factories.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/functional.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/permuted_chain_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/permuted_chain_env.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/policy.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/treesample/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/treesample/wrappers.py -------------------------------------------------------------------------------- /gfn_maxent_rl/envs/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/envs/wrappers.py -------------------------------------------------------------------------------- /gfn_maxent_rl/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/nets/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/nets/gnn.py -------------------------------------------------------------------------------- /gfn_maxent_rl/nets/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/nets/transformer.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/async_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/async_evaluation.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/beam_search.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/estimation.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/evaluations.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/exhaustive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/exhaustive.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/io.py -------------------------------------------------------------------------------- /gfn_maxent_rl/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/gfn_maxent_rl/utils/metrics.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tristandeleu/gfn-maxent-rl/HEAD/train.py --------------------------------------------------------------------------------