sfaira.estimators.EstimatorKerasEmbedding.predict

EstimatorKerasEmbedding.predict(batch_size: int = 128)

return the prediction of the model

Returns

prediction