gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-13 18:22:29 +02:00
Code Issues Projects Releases Wiki Activity

Include missing RegressionHead. (#263)

Browse Source
This commit is contained in:
Federico Andres Lois authored and Gal Leibovich committed 2019-04-16 15:24:06 +03:00
1 parent 20a8dea0dd
commit bdb9b224a8
1 file changed
+3 -1
Download Patch File Download Diff File Expand all files Collapse all files
rl_coach/architectures/tensorflow_components/heads/__init__.py
+3 -1
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'
] ]