def __init__(self, state_dim, control_dim, num_basis_functions, max_action=None):
MGPR.__init__(self,
np.random.randn(num_basis_functions, state_dim),
0.1*np.random.randn(num_basis_functions, control_dim)
)
for model in self.models:
model.kern.variance = 1.0
model.kern.variance.trainable = False
self.max_action = max_action
def __init__(self, X, Y, num_induced_points, name=None):
gpflow.Parameterized.__init__(self, name)
self.num_induced_points = num_induced_points
MGPR.__init__(self, X, Y, name)