garage.tf.baselines.gaussian_cnn_baseline module¶
Gaussian CNN Baseline.
-
class
GaussianCNNBaseline
(env_spec, subsample_factor=1.0, regressor_args=None, name='GaussianCNNBaseline')[source]¶ Bases:
garage.np.baselines.baseline.Baseline
GaussianCNNBaseline With Model.
It fits the input data to a gaussian distribution estimated by a CNN.
Parameters: - env_spec (garage.envs.env_spec.EnvSpec) – Environment specification.
- subsample_factor (float) – The factor to subsample the data. By default it is 1.0, which means using all the data.
- regressor_args (dict) – Arguments for regressor.
- name (str) – Name of baseline.
-
fit
(paths)[source]¶ Fit regressor based on paths.
Parameters: paths (dict[numpy.ndarray]) – Sample paths.
-
get_param_values
()[source]¶ Get parameter values.
Returns: A list of values of each parameter. Return type: List[np.ndarray]
-
get_params_internal
()[source]¶ Get the params, which are the trainable variables.
Returns: A list of trainable variables in the current variable scope. Return type: List[tf.Variable]