garage.np.baselines.baseline module

Base class for all baselines.

class Baseline(env_spec)[source]

Bases: abc.ABC

Base class for all baselines.

Parameters:env_spec (garage.envs.env_spec.EnvSpec) – Environment specification.
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]
log_diagnostics(paths)[source]

Log diagnostic information.

Parameters:paths (list[dict]) – A list of collected paths.
predict(path)[source]

Predict value based on paths.

Parameters:path (dict[numpy.ndarray]) – Sample paths.
Returns:Predicted value.
Return type:numpy.ndarray
set_param_values(flattened_params)[source]

Set param values.

Parameters:flattened_params (np.ndarray) – A numpy array of parameter values.