meds-torch

View on PyPIReverse Dependencies (0)

0.0.4 meds_torch-0.0.4-py3-none-any.whl

Wheel Details

Project: meds-torch
Version: 0.0.4
Filename: meds_torch-0.0.4-py3-none-any.whl
Download: [link]
Size: 113867
MD5: 0d1d0d38a961852c879e3bbd52c3f042
SHA256: 6fe8f10ee1ebadb0f601408160f7b54fb389b6ac4a90a4787cd77ea6d4ca29ff
Uploaded: 2024-09-20 02:53:51 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: meds-torch
Version: 0.0.4
Summary: A MEDS PyTorch Dataset, leveraging a on-the-fly retrieval strategy for flexible, efficient data loading.
Author-Email: Matthew McDermott <mattmcdermott8[at]gmail.com>, Nassim Oufattole <noufattole[at]gmail.com>, Teya Bergamaschi <teyabergamaschi[at]gmail.com>, Aleksia Kolo <aleksiakolo[at]gmail.com>
Project-Url: Homepage, https://github.com/oufattole/meds-torch
Project-Url: Issues, https://github.com/oufattole/meds-torch/issues
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.11
Requires-Dist: polars
Requires-Dist: pyarrow
Requires-Dist: nested-ragged-tensors
Requires-Dist: loguru
Requires-Dist: numpy
Requires-Dist: ml-mixins
Requires-Dist: torch
Requires-Dist: omegaconf
Requires-Dist: hydra-core
Requires-Dist: lightning
Requires-Dist: x-transformers (==1.34.0)
Requires-Dist: rich
Requires-Dist: hydra-colorlog
Requires-Dist: hydra-optuna-sweeper
Requires-Dist: wandb
Requires-Dist: torchmetrics
Requires-Dist: torchvision
Requires-Dist: transformers
Requires-Dist: ray[tune]
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: notebook; extra == "dev"
Requires-Dist: pdbpp; extra == "dev"
Requires-Dist: mkdocs (==1.6.0); extra == "docs"
Requires-Dist: mkdocs-gen-files (==0.5.0); extra == "docs"
Requires-Dist: mkdocs-get-deps (==0.2.0); extra == "docs"
Requires-Dist: mkdocs-git-authors-plugin (==0.9.0); extra == "docs"
Requires-Dist: mkdocs-git-revision-date-localized-plugin (==1.2.7); extra == "docs"
Requires-Dist: mkdocs-literate-nav (==0.6.1); extra == "docs"
Requires-Dist: mkdocs-material (==9.5.33); extra == "docs"
Requires-Dist: mkdocs-material-extensions (==1.3.1); extra == "docs"
Requires-Dist: mkdocs-section-index (==0.3.9); extra == "docs"
Requires-Dist: mkdocs-snippets (==1.3.0); extra == "docs"
Requires-Dist: mkdocstrings (==0.25.2); extra == "docs"
Requires-Dist: mkdocstrings-python (==1.10.8); extra == "docs"
Requires-Dist: pytest; extra == "tests"
Requires-Dist: pytest-cov[toml]; extra == "tests"
Requires-Dist: rootutils; extra == "tests"
Requires-Dist: pytest-sugar; extra == "tests"
Requires-Dist: pytest-instafail; extra == "tests"
Requires-Dist: pytest-xdist; extra == "tests"
Requires-Dist: sh; extra == "tests"
Requires-Dist: MEDS-transforms (==0.0.7); extra == "tests"
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: tests
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 7185 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.1.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
meds_torch/__init__.py sha256=6HAXfjMCm2QoERfqpZUmPW--cSxJMVZUg1u4yhRlxak 195
meds_torch/eval.py sha256=64dZg3cNAylT51-_AAyksti-kVuABJ0wObyV-GLsgIQ 2692
meds_torch/finetune.py sha256=r3S7vU9JlWZQhUNDBdJePzzvOUXx0Ca4E_bsbqDYWWI 5607
meds_torch/latest_dir.py sha256=Noyqpht652TdNe_Ny89JERMdpIjPzqDgNbgZ9OeNAEg 2777
meds_torch/train.py sha256=Lx1IpXRcz2I8s14KBypdRml_AQN9sXgTRc_rt-KzcaA 4590
meds_torch/tune.py sha256=ujcXOZKjKx9dTjBcmC7cLYbuHmpI3Fi1vq7UXUBx2Jk 6617
meds_torch/configs/__init__.py sha256=QK_jPpRhoEO3c6qTmz7DJxXQAxvGQRSDn_wWB69Af6g 81
meds_torch/configs/eval.yaml sha256=Jv0ecEInLxSVzZQ1KOi4e9SGfL-nIrE6PT3zA5rpmP8 362
meds_torch/configs/finetune.yaml sha256=jIwhCtoZjaLB8Xm6Dh7EQM1BEcZ3UvYibUaAAgMQLFo 1761
meds_torch/configs/latest_dir.yaml sha256=ueN0wfhRV8YNli91ulRpgzqUoewxVg-F35k-H8in64g 10
meds_torch/configs/train.yaml sha256=T6iH4lfLiwwRgEn0UYd_x1Dl4Xp-FaMZCu3RpPXql4U 1582
meds_torch/configs/callbacks/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/callbacks/default.yaml sha256=-4L23_jc6Zx4yYX1yb8tXM6V0Oi265m36VG9hWw9X8g 388
meds_torch/configs/callbacks/early_stopping.yaml sha256=_Xoilccs46yAhTaDeho05K1GoEpOq9JMtVrgrB7af3Q 1116
meds_torch/configs/callbacks/model_checkpoint.yaml sha256=iemBg2JgRrbhs7uJV4d1qNwJnQKzDWSuf-8-VGblmmI 1154
meds_torch/configs/callbacks/model_summary.yaml sha256=fGL4muU3V18vIqosZGcQwAEFsU04ttUxP0LdYxYHeI0 252
meds_torch/configs/callbacks/none.yaml sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/callbacks/ray.yaml sha256=6vY3QLcO-ucGgfZyw4fr5APcKw5LsMwzrpaM-91xlFU 103
meds_torch/configs/callbacks/rich_progress_bar.yaml sha256=ki7TYBXGTvyUQr5zej7uwPFViagvqNOVPE-AD8tcbYM 172
meds_torch/configs/callbacks/tune_default.yaml sha256=QtbN1ESb50xC_HJAI1Ny9Ow8PuUQIMLi4oMSpNsKOGo 372
meds_torch/configs/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/data/multiwindow_pytorch_dataset.yaml sha256=XVaxEVCfmUK_dP2SNb6zAm_UX3-o4zAOeKVXS1mF03s 126
meds_torch/configs/data/pytorch_dataset.yaml sha256=G80hFg6G3R27xlsjSsTOmuDhQgmL_szhLUAi156Bnts 1530
meds_torch/configs/data/random_windows_pytorch_dataset.yaml sha256=mXqp-Ng33_6d4bkwr1gBWdBscByFr9H2DTmbRH1yPEk 176
meds_torch/configs/debug/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/debug/default.yaml sha256=lOUqYBPxwV3o3j6f81BUHMS0sYZHaxC_4P3qX_hIuYw 899
meds_torch/configs/debug/fdr.yaml sha256=oFz5qVk8oI_6sbVDaFys1GPDVPMd5gUz5d_W9SDAzeY 120
meds_torch/configs/debug/limit.yaml sha256=DwQsu2BVzvchMTW3UbQOItKyl03T4hYWVus3qBVB8zc 218
meds_torch/configs/debug/overfit.yaml sha256=X0Ao60lKK12ld2Q5R-FnkvrCKPykHgFm4l-_s14URtU 204
meds_torch/configs/debug/profiler.yaml sha256=glSEp6FDsSkf_e2Pht7-EArBQvumNBf_TTVSzW2v--Q 177
meds_torch/configs/experiment/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/experiment/example.yaml sha256=g_-hbBgqGZPfShuxkTMxNDHi_0SNkzFkjIdmVWncWxY 607
meds_torch/configs/experiment/mimiciv.yaml sha256=sf28tAwY5EwfbF3MWUpPKn9S_hnlmWcEKl-yWYxrBdc 972
meds_torch/configs/extras/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/extras/default.yaml sha256=e7dBiY0r_SUy8SC79h5k2e2_K-y0yqpXIaQSOYsTq5U 232
meds_torch/configs/hparams_search/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/hparams_search/optuna.yaml sha256=dw05EDPT9r4AjuXZKYbepH_vHpHiNyCWjsnnrjwOBLg 1666
meds_torch/configs/hparams_search/ray_multiseed.yaml sha256=MU88o3B4bYphTMA_P6UMov7eygsNNROjVvQTQtpHcEM 362
meds_torch/configs/hparams_search/ray_tune.yaml sha256=12cMDqptl3t58ooJsp8seSB5Vq0FNaOH5V9SSRoqF7Q 649
meds_torch/configs/hydra/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/hydra/default.yaml sha256=gx2b_naNSRm9yHtaMMVoc-BYfbII_PTawtLCPPltHq0 579
meds_torch/configs/logger/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/logger/aim.yaml sha256=Fz4T9_CesjnqU7cxsyRjPorrGuQxyIi1oe34R8Zg18Y 1266
meds_torch/configs/logger/comet.yaml sha256=aUYk6WmaOI83pw2MYCeST_GiDi-9a6eyWIXu8MIEgMs 377
meds_torch/configs/logger/csv.yaml sha256=UZRXHmpznBHWsjEUhybz71dGohGp27AYOA_YjQojByU 162
meds_torch/configs/logger/many_loggers.yaml sha256=mI7UIHN7lbJGTs0D8hM0l5a8X5FKQTJJUH86VCUyM5k 118
meds_torch/configs/logger/mlflow.yaml sha256=1pyzb5jpLUxbcQMqtFu1ybirnnBF81uAqYrDRe9fL0s 339
meds_torch/configs/logger/neptune.yaml sha256=WakinZRrPe2yO_o4b_eAzAn4T9nMVbNdzaDv9TkRLew 277
meds_torch/configs/logger/tensorboard.yaml sha256=1T-f4n_ovBRA89O91iMJn61F_5GEDktcbcq4nHN8AVQ 263
meds_torch/configs/logger/wandb.yaml sha256=v8WTSm-02b-yIKW0De1XNxplWS6ERKmVhelicJ40YEw 527
meds_torch/configs/model/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/model/ebcl.yaml sha256=QBBv60LgN5ZMLsqtvylRIDA6dAan2hwyZRImwZyaQko 695
meds_torch/configs/model/eic_forecasting.yaml sha256=iMAUf2aJg6BeMaVYBklHDcMHkj-jri2u_5crAKG4j0M 689
meds_torch/configs/model/ocp.yaml sha256=tNRRJ5Ojtnn55kVaGQ0RmbveXBe0-KFHrxf2lUzBgs0 738
meds_torch/configs/model/supervised.yaml sha256=GiGMTzYBMPKQCNcfX-dAtAeR_qIesqc5CNyHpoEBjzE 653
meds_torch/configs/model/triplet_forecasting.yaml sha256=J8_imErVWyx3DK9Ua_AxeBGf0N3VEccvAwBeV85bnIM 664
meds_torch/configs/model/value_forecasting.yaml sha256=iuyuTGW4sPerqyXznYfXVofo2vML9qvFiJ0OH38xnzk 720
meds_torch/configs/model/backbone/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/model/backbone/default.yaml sha256=Bt2JTzmrVcPjSQD-6aF24lnlHRaC2T-pqvQ7tWpIWXw 233
meds_torch/configs/model/backbone/eic_lstm.yaml sha256=9iFuoxXQfnjUIsgoGdu2WP5-A9Ovt_9WVYtI1T4hfFM 138
meds_torch/configs/model/backbone/eic_transformer_decoder.yaml sha256=ZlJ4y9VVwDdfMYtHYL-VC_UtmimpPAt6xygQKT8Eh1g 809
meds_torch/configs/model/backbone/eic_transformer_encoder.yaml sha256=NXPgdIRRfBQ9oamarPHPdeJY_8DaCt-nwKC3LWZOojo 839
meds_torch/configs/model/backbone/eic_transformer_encoder_attn_avg.yaml sha256=WjO8mF9v3lmwkTR04kdCq1EuP1RIhSubhwlVmSjqHn4 844
meds_torch/configs/model/backbone/triplet_lstm.yaml sha256=ms204Kfp5AwROSGF1yhEwxov-mM3_thD56Pa-4zD0B4 119
meds_torch/configs/model/backbone/triplet_transformer_decoder.yaml sha256=-kBLGyOFfWJrwVFCETrG7T23lA_7JE1CPS1WSMi4rbc 882
meds_torch/configs/model/backbone/triplet_transformer_encoder.yaml sha256=70VdWWaLsVnIfyWu6L9YZmxe9DdO_16_ak9AXrBPCoE 900
meds_torch/configs/model/backbone/triplet_transformer_encoder_attn_avg.yaml sha256=WNMbSI_gszv-py_N1r0_acqjEyVy8LE4PdFmSsgaOus 883
meds_torch/configs/model/input_encoder/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/model/input_encoder/eic_encoder.yaml sha256=ZHvg7AKFcaw7iDtrp_M_meONIcyEOqXiq7gmRf9kNUY 190
meds_torch/configs/model/input_encoder/text_code_encoder.yaml sha256=pMTS4xPQHcr3TvurZNf1eGpzDnrnzTrwwE1LGo7TCac 1193
meds_torch/configs/model/input_encoder/triplet_encoder.yaml sha256=V4jyyU8AR68KMHUMM_Ji5jkEq3S5_4WvY6ap5Cf4dnc 215
meds_torch/configs/model/input_encoder/triplet_prompt_encoder.yaml sha256=88N7zvR_u5bN8kL6cq-OL-EyaC_iaDABLRoZR4pDiE8 228
meds_torch/configs/paths/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/paths/default.yaml sha256=iCQ3pF8O5ylKs4Uz7Z3FjhMLMo2ThMK81LS6VZD4_JM 477
meds_torch/configs/trainer/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/configs/trainer/cpu.yaml sha256=qVDce-EC47A7YLAHhXjBnhdFlxXXjdWstzsVz6V-H-4 51
meds_torch/configs/trainer/ddp.yaml sha256=6UKVbyXuhKwlHvzSbHNdou5z1T-F6U1Fc87_smnsCSA 100
meds_torch/configs/trainer/ddp_sim.yaml sha256=I9-Sg-Y4GyAhrR5aYH4eMJhPe2FqhbH8jpUOMn5aWWI 115
meds_torch/configs/trainer/default.yaml sha256=4VQDurcRnffUAlwUDKya9_LSJdljGiEiR__ZKV_TXew 454
meds_torch/configs/trainer/gpu.yaml sha256=58LTGS9Zoa09pOguMEuVqM2YYoUL61wr6CxsDOaI4wo 51
meds_torch/configs/trainer/mps.yaml sha256=DoRNhWQIQucCIqX8wsbqUvwdn_D7tvqNZLjJv55zC-o 51
meds_torch/configs/trainer/ray.yaml sha256=a22i6DOMXL44ENmcFWI22XSzX9AA5BqUb2oZxW9-LhY 118
meds_torch/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/data/datamodule.py sha256=xP1djS2OJic9zL-l8BIEM6MHQ-9ZNZ5QFz-51cGa6Ik 6590
meds_torch/data/components/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
meds_torch/data/components/multiwindow_pytorch_dataset.py sha256=lUndw9BFJbxI0FPJmV2Gb5bAcd7zzwYCORjZ1xxxrds 13745
meds_torch/data/components/pytorch_dataset.py sha256=2qqm9Gg-s0N9nYUkC_CEko3tRMxm-9nnS7vaYvW9yL8 70514
meds_torch/data/components/random_windows_pytorch_dataset.py sha256=hy0n8uD-unSyEbKJg-RsUzBVcTMTEMmSbvNRdb0QPXg 6389
meds_torch/input_encoder/README.md sha256=nA0-QU1AqeBcRKa6JVqu9IHLL8AJ5fygyft0Uzi0gVs 1025
meds_torch/input_encoder/__init__.py sha256=6nP0Ue8mvESNPRuUCKXqPg6clf-RabVyfUBh74yHrlY 98
meds_torch/input_encoder/eic_encoder.py sha256=S2aXmhBsxyfeURdStp45cVckp6-oerJBLxOtLOPFWEs 898
meds_torch/input_encoder/text_encoder.py sha256=yXxkc1cVw-JPryxRT_VnyH_4EI5pq8ebeoQc0Q63b84 5252
meds_torch/input_encoder/triplet_encoder.py sha256=WJlDUP9nhVhHGUvJDLe9-zFTSUYz7Y39gFGw3MfpDPs 3063
meds_torch/input_encoder/triplet_prompt_encoder.py sha256=Vwqw0-gqo0qytGNarq1bG0TL1yvnwoCHvUXwl6yAoBU 2535
meds_torch/models/__init__.py sha256=DJROMpzUbR9x8qXB4y_e_y7ZMKbWzxqYEhOaCMdt0bY 236
meds_torch/models/base_model.py sha256=c8RM3T03IXrbyy7E6r24TN-TuCeN4F7VYP35fqeB6RE 1656
meds_torch/models/ebcl_model.py sha256=btCvKSlHKgNrUnJQErlxX6J3r4BXD7RnWe0QuM3U6M0 8463
meds_torch/models/eic_forecasting.py sha256=FeRrIyBAjPhvbVvtaAKlVfs7fu73_xAzV4XMzgqLMv8 5192
meds_torch/models/ocp_model.py sha256=cpUZkA-MkieUDh5AijCJMS8QaqyyRSw_qMg272lAB1I 8177
meds_torch/models/supervised_model.py sha256=3F98FhOvjgedhXvlWoA4E_TTNGj0sqHb4hlBEYf_Dvo 5729
meds_torch/models/triplet_forecasting.py sha256=Ao4EfYpmO_8gb3BgoF4FwK0-mi6z7btpdXC9kDpXx6o 15759
meds_torch/models/utils.py sha256=J5EV9khDXNjneKdell-3N0zfaoVVT6a3Y78lumv_K4g 99
meds_torch/models/value_forecasting.py sha256=3yKdRjbguFIG_kF0XdgQf5oNFW28YlF338sC4uTuXa4 5889
meds_torch/models/components/__init__.py sha256=sTFwdxRFcQO4PgycFpqsT9TjBgQRXg9I2SFJTx2Jpv0 203
meds_torch/models/components/lstm.py sha256=_GGqM86O56qguPsq0cRtF8KMMCOo1Al7i2a9HjnBQeA 1364
meds_torch/models/components/transformer_decoder.py sha256=hhHyRpB6J63EzkABkI22db9STFgGyjU_y4n-ZmUb0Z4 1351
meds_torch/models/components/transformer_encoder.py sha256=RCCj05GK6teGKfAid2JBbDiVxCHdWEh4KVg_CwwVviU 1087
meds_torch/models/components/utils.py sha256=kzJHrTA71ALwEPzVZBFeqkfRHEq4WaJRLv7cjnPQVB0 1143
meds_torch/utils/__init__.py sha256=aWjOijaHpxAG4n4g-BZ9ycbe7eLUBxsavWal0FYhGPg 399
meds_torch/utils/custom_normalization.py sha256=j3284n-LRLvnX-aw3lp84nJd2OsaoStmWNWcBYglIFw 25736
meds_torch/utils/custom_time_token.py sha256=EkOO5qQAfGPw8Q9dryiyACE_E7nhJ7Tw9oYLubiFsnA 22312
meds_torch/utils/custom_tokenization.py sha256=1pnCzzcKnejYgplWrZFWBgu1urj4Pc16EGAqOYP-khc 12873
meds_torch/utils/instantiators.py sha256=qoLFoN2z1RqUVdF8hfcC5PSStCEk-owTTCLxdoycVaY 1756
meds_torch/utils/logging_utils.py sha256=ouuMd8-_isxCY9pjeIC1PeK9dKBgYTrDXmJsG9sf_7o 1717
meds_torch/utils/module_class.py sha256=Z3Fx_uhlcs7aaXPEeGZQbfZQiDOqjeBTbUT8j37bF1c 272
meds_torch/utils/pylogger.py sha256=Kd9_ykPi-Rh9pyHda9MvYavZ6XDvLLlo8N1GoY4-4x0 2433
meds_torch/utils/resolvers.py sha256=r-VZEjTAVVCypezYyVorpwlnHg9k3jovIxYYzdNXm1M 1378
meds_torch/utils/rich_utils.py sha256=9VkTHwkcUHRGEgJCGM6Z11Y54eDofDoWsQGVgmEUPSI 3304
meds_torch/utils/utils.py sha256=NyabQznrx4igmYLyAXmuicU54xYUunAGhr6zwkh6KuA 4197
meds_torch-0.0.4.dist-info/LICENSE sha256=vfHx9yV2jAluYuE2-8ogR6pz5rxVy4oLTJ6yy3Tm3hg 1110
meds_torch-0.0.4.dist-info/METADATA sha256=Kol190SuPQ2Al2POJfSbAQhEpoPjVEEYFm7PTnPoTDA 9680
meds_torch-0.0.4.dist-info/WHEEL sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ 91
meds_torch-0.0.4.dist-info/entry_points.txt sha256=vAnZoaRiWbXdLLZlrIOgKKbvgNxaS9v0hM6aCuGrSrU 235
meds_torch-0.0.4.dist-info/top_level.txt sha256=iocVu6-ElKqN333cyG9yGdO0zuXHUEWn7Pai4G3jomI 11
meds_torch-0.0.4.dist-info/RECORD

top_level.txt

meds_torch

entry_points.txt

meds-torch-eval = meds_torch.eval:main
meds-torch-finetune = meds_torch.finetune:main
meds-torch-latest-dir = meds_torch.latest_dir:main
meds-torch-train = meds_torch.train:main
meds-torch-tune = meds_torch.tune:main