cellflow.solvers.OTFlowMatching.get_condition_embedding

OTFlowMatching.get_condition_embedding(condition, return_as_numpy=True)

Get learnt embeddings of the conditions.

Parameters:

  • condition (dict[str, ndarray[tuple[Any, ...], dtype[float64]]]) – Conditions to encode.

  • return_as_numpy – Whether to return the embeddings as numpy arrays.

Return type:

ndarray[tuple[Any, ...], dtype[float64]]

Returns:

: Mean and log-variance of encoded conditions.