b075341dfc3e719655647a97312c11426a285c26,density_estimator/LSCDE.py,LSConditionalDensityEstimation,fit,#LSConditionalDensityEstimation#Any#Any#,32

Before Change

  def fit(self, X, Y, **kwargs):
    // assert that both X an Y are 2D arrays with shape (n_samples, n_dim)
    if X.ndim == 1:
      X = np.expand_dims(X, axis=1)
    if Y.ndim == 1:
      Y = np.expand_dims(Y, axis=1)

    self.ndim_y, self.ndim_x = Y.shape[1], X.shape[1]
    assert X.ndim == Y.ndim == 2

    // define the full model

After Change

  def fit(self, X, Y, **kwargs):
    // assert that both X an Y are 2D arrays with shape (n_samples, n_dim)
    X, Y = handle_input_dimensionality(X, Y)
    self.ndim_y, self.ndim_x = Y.shape[1], X.shape[1]


    // define the full model

Instances