garage.np.baselines.linear_feature_baseline module

A linear value function (baseline) based on features.

class LinearFeatureBaseline(env_spec, reg_coeff=1e-05, name='LinearFeatureBaseline')

Bases: garage.np.baselines.baseline.Baseline

A linear value function (baseline) based on features.

Parameters:

• env_spec (garage.envs.env_spec.EnvSpec) – Environment specification.
• reg_coeff (float) – Regularization coefficient.
• name (str) – Name of baseline.

fit(paths)

Fit regressor based on paths.

Parameters:paths (list[dict]) – Sample paths.

get_param_values()

Get parameter values.

Returns:A list of values of each parameter. Return type:List[np.ndarray]

predict(path)

Predict value based on paths.

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

set_param_values(flattened_params)

Set param values.

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