sfaira.estimators.EstimatorKerasCelltype.predict

EstimatorKerasCelltype.predict(batch_size: int = 128, max_steps: int = inf)

Return the prediction of the model

Parameters

  • batch_size – Batch size for evaluation.

  • max_steps – Maximum steps before evaluation round is considered complete.

Returns

Prediction tensor.