garage.np.baselines.linear_multi_feature_baseline

Linear Multi-Feature Baseline.

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

Bases: garage.np.baselines.LinearFeatureBaseline

Inheritance diagram of garage.np.baselines.linear_multi_feature_baseline.LinearMultiFeatureBaseline

A linear value function (baseline) based on features.

Parameters
  • env_spec (garage.envs.env_spec.EnvSpec) – Environment specification.

  • reg_coeff (float) – Regularization coefficient.

  • features (list[str]) – Name of features.

  • name (str) – Name of baseline.

get_param_values()

Get parameter values.

Returns

A list of values of each parameter.

Return type

List[np.ndarray]

set_param_values(flattened_params)

Set param values.

Parameters

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

fit(paths)

Fit regressor based on paths.

Parameters

paths (list[dict]) – Sample paths.

predict(paths)

Predict value based on paths.

Parameters

paths (list[dict]) – Sample paths.

Returns

Predicted value.

Return type

numpy.ndarray