0911c4f8477ee084c4346616d10387d56a8c58f1,cde/density_simulation/GMM.py,GaussianMixture,simulate_conditional,#GaussianMixture#Any#,121

Before Change

    

    W_x = self._W_x(X)
    Y = np.zeros(shape=(X.shape[0], self.ndim_y))
    for i in range(X.shape[0]):
      discrete_dist = stats.rv_discrete(values=(range(self.n_kernels), W_x[i,:]))
      idx = discrete_dist.rvs()
      Y[i, :] = self.gaussians_y[idx].rvs()
    assert X.shape[0] == Y.shape[0]
    return X, Y

  def _draw_from_discrete(self, w_x):

After Change

    X = self._handle_input_dimensionality(X)

    if np.all(np.all(X == X[0, :], axis=1)):
      return self._simulate_rows_same(X)
    else:
      return self._simulate_rows_individually(X)

Instances