gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

new feature - implementation of Quantile Regression DQN (https://arxiv.org/pdf/1710.10044v1.pdf)

Browse Source 
API change - Distributional DQN renamed to Categorical DQN
This commit is contained in:
Itai Caspi
2017-11-01 15:09:07 +02:00
parent 1ad6262307
commit a8bce9828c
10 changed files with 157 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
architectures/tensorflow_components/general_network.py
View File

@@ -80,7 +80,8 @@ class GeneralTensorFlowNetwork(TensorFlowArchitecture):
OutputTypes.NAF: NAFHead,
OutputTypes.PPO: PPOHead,
OutputTypes.PPO_V : PPOVHead,
OutputTypes.DistributionalQ: DistributionalQHead
OutputTypes.CategoricalQ: CategoricalQHead,
OutputTypes.QuantileRegressionQ: QuantileRegressionQHead
}
return output_mapping[head_type](self.tp, head_idx, loss_weight, self.network_is_local)