garage.np.baselines.linear_feature_baseline

A linear value function (baseline) based on features.

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

Bases: garage.np.baselines.baseline.Baseline

Inheritance diagram of garage.np.baselines.linear_feature_baseline.LinearFeatureBaseline

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.

get_param_values()[source]

Get parameter values.

Returns

A list of values of each parameter.

Return type

List[np.ndarray]

set_param_values(flattened_params)[source]

Set param values.

Parameters

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

fit(paths)[source]

Fit regressor based on paths.

Parameters

paths (list[dict]) – Sample paths.

predict(paths)[source]

Predict value based on paths.

Parameters

paths (list[dict]) – Sample paths.

Returns

Predicted value.

Return type

numpy.ndarray