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

Include missing RegressionHead. (#263)

Browse Source
This commit is contained in:
Federico Andres Lois
2019-04-16 09:24:06 -03:00
committed by Gal Leibovich
parent 20a8dea0dd
commit bdb9b224a8
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/architectures/tensorflow_components/heads/__init__.py
View File

@@ -12,6 +12,7 @@ from .quantile_regression_q_head import QuantileRegressionQHead
from .rainbow_q_head import RainbowQHead from .rainbow_q_head import RainbowQHead
from .v_head import VHead from .v_head import VHead
from .acer_policy_head import ACERPolicyHead from .acer_policy_head import ACERPolicyHead
from .cil_head import RegressionHead
__all__ = [ __all__ = [
'CategoricalQHead', 'CategoricalQHead',
@@ -27,5 +28,6 @@ __all__ = [
'QuantileRegressionQHead', 'QuantileRegressionQHead',
'RainbowQHead', 'RainbowQHead',
'VHead', 'VHead',
'ACERPolicyHead' 'ACERPolicyHead',
'RegressionHead'
] ]