Utils

Project root

Returns:

Name Type Description
str Path

Path to the root of the project.

Set seed

Sets the numpy and torch seeds.

Parameters:

Name Type Description Default
k

seed.

required

Plot trajectory

Function to plot the time evolution of the state of a cell.

Parameters:

Name Type Description Default
trajectory torch.Tensor

States of the cell observed at the different time points. Shape (n_time_points, 1, n_genes).

required
time_points list or 1D numpy.array

Times of the observations.

None
legend bool

Whether to add a legend to the plot.

True

Raises:

Type Description
RuntimeWarning

If the trajectory contains more than one cell. Only the trajectory of the first cell is plotted.