import numpy as np
from garage.np.baselines.baseline import Baseline
[docs]class ZeroBaseline(Baseline):
def __init__(self, env_spec):
pass
[docs] def get_param_values(self, **kwargs):
return None
[docs] def set_param_values(self, val, **kwargs):
pass
[docs] def fit(self, paths):
pass
[docs] def predict(self, path):
return np.zeros_like(path['rewards'])
[docs] def predict_n(self, paths):
return [np.zeros_like(path['rewards']) for path in paths]