A neural ode solution for imputing trajectories between pointclouds.
Project description
This library runs code associated with the TrajectoryNet paper [1].
Installation
TrajectoryNet is available in pypi. Install by running the following
pip install TrajectoryNet
This code was tested with python 3.7 and 3.8.
Example
Basic Usage
Run with
python -m TrajectoryNet.main --dataset SCURVE
To use a custom dataset expose the coordinates and timepoint information according to the example jupyter notebooks in the /notebooks/ folder.
TrajectoryNet requires the following:
An embedding matrix titled [embedding_name] (Cells x Dimensions)
A sample labels array titled sample_labels (Cells)
(Optionally) a delta embedding representing RNA velocity titled delta_[embedding_name] (Cells x Dimensions)
To run TrajectoryNet with a custom dataset use:
python -m TrajectoryNet.main --dataset [PATH_TO_NPZ_FILE] --embedding_name [EMBEDDING_NAME]
python -m TrajectoryNet.eval --dataset [PATH_TO_NPZ_FILE] --embedding_name [EMBEDDING_NAME]
See notebooks/EB-Eval.ipynb for an example on how to use TrajectoryNet on a PCA embedding to get trajectories in the gene space.
References
[1] Tong, A., Huang, J., Wolf, G., van Dijk, D., and Krishnaswamy, S. TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular Dynamics. In International Conference on Machine Learning, 2020. arxiv ICML
—
If you found this library useful, please consider citing:
@inproceedings{tong2020trajectorynet, title = {TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular Dynamics}, shorttitle = {{{TrajectoryNet}}}, booktitle = {Proceedings of the 37th International Conference on Machine Learning}, author = {Tong, Alexander and Huang, Jessie and Wolf, Guy and {van Dijk}, David and Krishnaswamy, Smita}, year = {2020} }
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for TrajectoryNet-0.2.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2d912fe0d63f57ae7a2dcf53a22b3d5de26abf850f5d5158bf15070f4cfd261c |
|
MD5 | c70cc9567759ed473b784ab35a28a295 |
|
BLAKE2b-256 | ad27e1a922f43c1ca4f1c4aba55adcda33c8791ae68a14d2cae7d5a2ad4be87e |