cellrank.ul.models.SKLearnModel.predict

SKLearnModel.predict(x_test=None, key_added='_x_test', **kwargs)[source]

Run the prediction.

Parameters
  • x_test (Optional[ndarray]) – Array of shape (n_samples,) used for prediction. If None, use x_test.

  • key_added (str) – Attribute name where to save the x_test for later use. If None, don’t save it.

  • **kwargs – Keyword arguments for underlying model’s prediction method.

Returns

Updates and returns the following:

  • y_test - Prediction values of shape (n_samples,) for x_test.

Return type

numpy.ndarray