A python toolkit for Trajectory Deep Learning.
TrajDL提供了轨迹数据挖掘领域中的多个SOTA深度学习模型的实现,为研究人员、工程师提供易用、高效、可靠的开发工具,可以快速开展实验和应用开发。TrajDL有几个关键特性:
-
基于Arrow,Pytorch和Lightning
TrajDL的数据部分构建在Arrow之上,模型部分构建在Pytorch之上,训练与验证流程构建在Lightning之上,充分结合各个框架工具的优势。 -
高效的工具
TrajDL提供了高效的工具,比如高效的Dataset,Tokenizer,GridSystem。出色的零拷贝特性可以显著降低数据的处理时间,节省内存使用。高效的Tokenizer和GridSystem可以随时转换数据,无需预先处理数据。 -
可扩展性
TrajDL高度模块化,不会约束用户的代码,用户可以随时从TrajDL里面取出自己需要使用的工具。TrajDL还打通了与Polars,Pandas,PyArrow等工具的接口,用户使用常用的科学计算工具处理后的数据可以轻松导入到TrajDL的数据体系。另外TrajDL同时支持API与配置文件两种方式开展实验与开发,尽可能提升用户体验。 -
包含SOTA模型的实验复现脚本
TrajDL提供了SOTA模型的复现脚本,用户可以通过脚本重现论文内的实验结果,部分场景下TrajDL具备比论文场景更优的效果。
文档 📕
简体中文文档参阅:简体中文文档
English documentation will be provided in subsequent versions.
Benchmark 🚀
scripts/benchmark目录下存储了TrajDL提供的benchmark复现脚本,针对各个论文使用TrajDL进行了实验复现。
License
本项目使用Apache License 2.0,详见LICENSE。
Metadata
Release files for trajdl 0.1.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| trajdl-0.1.0.tar.gz | 961.5 kB | Details |
Built distributions (wheels)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| trajdl-0.1.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.12 | CPython 3.12 | Linux glibc 2.17+ x86-64 | Details |
| trajdl-0.1.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.11 | CPython 3.11 | Linux glibc 2.17+ x86-64 | Details |
| trajdl-0.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.10 | CPython 3.10 | Linux glibc 2.17+ x86-64 | Details |
Total release size: 1.8 MB
Release files / trajdl-0.1.0.tar.gz
| Download URL | trajdl-0.1.0.tar.gz |
|---|---|
| Size | 961.5 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
c8751a6871401bc708b0c0e5c83b7eed0a85b1457f72579126659fd6175211f6
|
|
BLAKE2b-256 checksum How to use checksums |
91b4ec7ed8325f99d5270f35b931bae7e80f5dbb8df4b94416efbe3af6da3a44
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.12.4
|
Release files / trajdl-0.1.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | trajdl-0.1.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 265.1 kB |
| Tags | CPython 3.12 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
1970a801f09f6d759938edfe86a6c0d216a21924467a133255d7981987607106
|
|
BLAKE2b-256 checksum How to use checksums |
b4c6a760d48d6b37ea7948bdfd6a5d6c255d5c8d29809db9426f69c974c26363
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.12.4
|
Release files / trajdl-0.1.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | trajdl-0.1.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 265.2 kB |
| Tags | CPython 3.11 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
9f6b15a8267c930d5c5b26c191c1d6fa9bad379b0aeb0cb83551d0943507aa1f
|
|
BLAKE2b-256 checksum How to use checksums |
5f14234628966b5ad5b419f429c3c1d62903900d5d1a66e7f1faa154e7a8eb3b
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.12.4
|
Release files / trajdl-0.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | trajdl-0.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 264.2 kB |
| Tags | CPython 3.10 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
996c8604b0ee6c47bbc4807ce993c512cfe4d299d91794e1f2bfe223a92dc811
|
|
BLAKE2b-256 checksum How to use checksums |
e2348301e70235b6a500ef73e99f71fb5b95300b257ea337570a916c7b1691e6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.12.4
|