Lines Matching refs:s_indices
299 def __init__(self, R, Q, beta, s_indices=None, a_indices=None): argument
331 if s_indices is None:
335 if not (len(s_indices) == self.num_sa_pairs and
342 self.s_indices = np.asarray(s_indices)
345 if _has_sorted_sa_indices(self.s_indices, self.a_indices):
347 _generate_a_indptr(self.num_states, self.s_indices,
353 (np.arange(self.num_sa_pairs), (s_indices, a_indices))
367 self.s_indices = _s_indices
399 self.s_indices, self.a_indices = None, None
531 R[self.s_indices, self.a_indices] = self.R
534 _fill_dense_Q(self.s_indices, self.a_indices,
537 _fill_dense_Q(self.s_indices, self.a_indices, self.Q, Q)