catenets

View on PyPIReverse Dependencies (0)

0.2.3 catenets-0.2.3-py3-none-any.whl
catenets-0.2.3-py3-none-macosx_10_14_x86_64.whl

Wheel Details

Project: catenets
Version: 0.2.3
Filename: catenets-0.2.3-py3-none-macosx_10_14_x86_64.whl
Download: [link]
Size: 130151
MD5: c6d794dce6ec574b0fb10c02ee36b11e
SHA256: beee8538fa5e4c18f5b2e78b850fabea0d5091e7950f4fed8441481d08738569
Uploaded: 2023-03-01 08:30:16 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: catenets
Version: 0.2.3
Summary: Conditional Average Treatment Effect Estimation Using Neural Networks
Author: Alicia Curth
Author-Email: amc253[at]cam.ac.uk
License: BSD-3-Clause
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: OS Independent
Platform: any
Requires-Python: >=3.7
Requires-Dist: gdown
Requires-Dist: jax (>=0.3.16)
Requires-Dist: loguru (>=0.5.3)
Requires-Dist: numpy (>=1.20)
Requires-Dist: pandas (>=1.3)
Requires-Dist: pytest (>=6.2.4)
Requires-Dist: requests
Requires-Dist: scikit-learn (>=0.24.2)
Requires-Dist: scipy (>=1.2)
Requires-Dist: torch (>=1.9)
Requires-Dist: importlib-metadata; python_version < "3.8"
Requires-Dist: jaxlib (>=0.3.14); sys_platform != "win32"
Requires-Dist: bandit; extra == "testing"
Requires-Dist: black; extra == "testing"
Requires-Dist: catboost; extra == "testing"
Requires-Dist: flake8; extra == "testing"
Requires-Dist: pre-commit; extra == "testing"
Requires-Dist: pytest; extra == "testing"
Requires-Dist: sklearn; extra == "testing"
Requires-Dist: xgboost; extra == "testing"
Requires-Dist: setuptools; extra == "testing"
Requires-Dist: pytest-cov; extra == "testing"
Requires-Dist: jupyter; extra == "testing"
Requires-Dist: notebook; extra == "testing"
Requires-Dist: py; extra == "testing"
Provides-Extra: testing
Description-Content-Type: text/markdown; charset=UTF-8
License-File: LICENSE
[Description omitted; length: 6191 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.38.4)
Root-Is-Purelib: true
Tag: py3-none-macosx_10_14_x86_64

RECORD

Path Digest Size
catenets/__init__.py sha256=tY1m39x9MVufVGXzpYXmD5lys8UoR_m8gwu7qMiTBqA 139
catenets/logger.py sha256=43WhjjqzxkEK4AGghSjt8uT9m98nK6ZnJmBWI9T0P98 2771
catenets/version.py sha256=PNiDER4qM19h9zdsdfgKt2_dT4WgYK7EguJ8RU2qA_g 22
catenets/datasets/__init__.py sha256=1Et08g4BTlzOJWNxw3yo38n2enO2adqnNmsTY9l0-QY 975
catenets/datasets/dataset_acic2016.py sha256=Iz6RYYFmWjdtX6OX27c1PbzC5vUPJ2RpkjlwWa38H-M 10034
catenets/datasets/dataset_ihdp.py sha256=w7ovwFaXJWeMqpGFlHQX-YNPr8jvzPdjSu7RvkuP9P8 6473
catenets/datasets/dataset_twins.py sha256=NtM1pPJNBoXNgyjfZS7SodIS4Y2A8TCO8AGBJMyn-bI 6713
catenets/datasets/network.py sha256=E3rGjBQFecg6Fx8ULNUmGGCAO95NOCdjdvEzZYMkcks 3289
catenets/experiment_utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
catenets/experiment_utils/base.py sha256=yNLlwj2O5f0t8O7qBOVdgxr1XAi2skdgjg44aPQO6GA 3675
catenets/experiment_utils/simulation_utils.py sha256=U_IZ7VzGN4lV5hFRPhLJCJGOrzBSr2QWBmJBdC958xI 9217
catenets/experiment_utils/tester.py sha256=WNf-0xSm_W6Cdc0muF8GVi1tXlCIw_aI3EVqyRw37M0 1951
catenets/experiment_utils/torch_metrics.py sha256=WbOcTcr05b5fQARKwXKIDPe7EHnYSndkeRaHMfVXfSk 922
catenets/models/__init__.py sha256=FS1x_eSZZerHdEXOgvN0XOxwIK2uebaZK_2noPF0-8Y 234
catenets/models/constants.py sha256=40WRT87T4rqbtjQsqL5P-P_SGl2qrPjDhdpqmkuYyvo 1114
catenets/models/jax/__init__.py sha256=YsPTVNxHeLEyb6W9bNKhxKS8oa0AIiHbv7jMcL-bxqU 2044
catenets/models/jax/base.py sha256=ek823L4jR3V5Yt_uSRSNz3HDIqRWLzorlxLnfvMD4m0 13743
catenets/models/jax/disentangled_nets.py sha256=ZOpTbPr6ngF_nLySK7ok5A480_agIkBNP5SVjDHL_EQ 19103
catenets/models/jax/flextenet.py sha256=S-A0TfrsPp3dOYAYBbH4js1QyHqWBn2au-VyQWt2Sag 33263
catenets/models/jax/model_utils.py sha256=F7seC6P5CkaPNxXcQP9kk0frJRUqIoH03wd91DA8DbQ 2905
catenets/models/jax/offsetnet.py sha256=kU1H7vMdxnHvcPCRRGfM6zPaYg5adqXd6rrwwJQOkXk 12103
catenets/models/jax/pseudo_outcome_nets.py sha256=IhAWHiFLIPzP9Dqjiqvho3Eqx0TbwHPRa7CxnjB1R98 28778
catenets/models/jax/representation_nets.py sha256=oXxSdOFVqON1OzAxP1vVeeIbrCh7MgybYWuBII_nB3Y 31554
catenets/models/jax/rnet.py sha256=cZyvd5xdhfp6cI5djzjvUEQbKK82jD3Hh_afwtSUANU 20157
catenets/models/jax/snet.py sha256=rTQv_WUSKLXL2tgpikiQQgNB6TiWvDrfZBDHLHDbAP0 34017
catenets/models/jax/tnet.py sha256=jdw3COWfnulYKzF2O-6xSEEppyI5z91-rIlf4-q-XHA 14991
catenets/models/jax/transformation_utils.py sha256=IBQjgkDZv3rl8mPaW6GYDK9YUSWLDNcaqZksKZJ5_qA 4044
catenets/models/jax/xnet.py sha256=r_L_2Wj1ieLPQjy8CfcUL1PxHg1q3w2cnMWEVrq5dDs 16947
catenets/models/torch/__init__.py sha256=5I5ckB0EXnLZufmSiehoY9k5nMgCfTQ9hFD7O60QEag 562
catenets/models/torch/base.py sha256=n02OUv17FOvJTNSoOSGpggKprXTMGaZGOJgOQjcSN3A 20487
catenets/models/torch/flextenet.py sha256=yG8R5y2UK1EfV60ZKANnuvgo-Sbj0vntxIgjSBOOgds 22738
catenets/models/torch/pseudo_outcome_nets.py sha256=2uxAduJ3z8qJafnQqqjnLmhY_xiglKvAsidLe2d8d84 22113
catenets/models/torch/representation_nets.py sha256=UdysuUldUxw1y-81ftsGHC1ggVLsuetxt2MC1rx2iGg 14345
catenets/models/torch/slearner.py sha256=oL3PhLkPERr2UZRkHAH3B3m7zojLGEz5RWiVg_Vtk80 7539
catenets/models/torch/snet.py sha256=mjEBRdoJZZZW9otFgqPjcHKA8gWNfV4-egM58Cfs3tQ 19368
catenets/models/torch/tlearner.py sha256=87sGKPRv4k4D0cPQ6J-DF-iLgoQTpf904Z0XoYgAvEE 5095
catenets/models/torch/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
catenets/models/torch/utils/decorators.py sha256=DYakg-rztNiedUBo55IySr08D10Aj08hcMWlqeOR0aA 1100
catenets/models/torch/utils/model_utils.py sha256=nEizSjyWufhC4tdiNbJVd5dVWhXUMEQP09ZVhBdWn8g 3082
catenets/models/torch/utils/transformations.py sha256=0zNDC3EhI4UA1sfCMAmg56a1ht7CaQAdov5Z2B7UzEo 4388
catenets/models/torch/utils/weight_utils.py sha256=wKMOgCgRO90OJA6uAHVA-Rs7ybVc_Vu5ss67VubcoR4 2188
docs/conf.py sha256=TWddDgR_Dt2--xcF2Re2JIvFnJompYBZuZmZlzCQ15g 2599
experiments/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
experiments/experiments_AISTATS21/ihdp_experiments.py sha256=DCd47HRDnC7a6Pbmn3NKePqmziBLmVplQ9VIhqUxE9E 3765
experiments/experiments_AISTATS21/simulations_AISTATS.py sha256=G1MaI9SQJhuxVZRuRXbV6ym19RGWrHLx6Ma7jKJoRQc 13756
experiments/experiments_benchmarks_NeurIPS21/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
experiments/experiments_benchmarks_NeurIPS21/acic_experiments_catenets.py sha256=9ztJx0_pqIWjHt6HjsMq1Yrtx7OpeMgwVQV8PbNr0T4 2917
experiments/experiments_benchmarks_NeurIPS21/ihdp_experiments_catenets.py sha256=B9cAJqjzbx9_W61Tgtn4iGW-LZFxCFE-6ZApjiVEfdg 3620
experiments/experiments_benchmarks_NeurIPS21/twins_experiments_catenets.py sha256=UeuQMrh1jWkT06Wml-xS9O12BiEJo35Aef5FLMmF8dk 5880
experiments/experiments_inductivebias_NeurIPS21/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
experiments/experiments_inductivebias_NeurIPS21/experiments_AB.py sha256=TPH4SWn0PSRY8qM6pnCaxKihSqy9oSod8R0AG1OQwgM 13513
experiments/experiments_inductivebias_NeurIPS21/experiments_CD.py sha256=QZnYhzBfpbUL5_zcuJyeiAIUY8L2ictpp__Bgdycyy8 3559
experiments/experiments_inductivebias_NeurIPS21/experiments_acic.py sha256=mBcDhnd0iX59X-iWH7RrVNSkrF1okOB8Bh-JaroMZhI 3919
experiments/experiments_inductivebias_NeurIPS21/experiments_twins.py sha256=V_v-axh--TL2FVfJqQTGGvTOdM-PvOKg7GL4SlvrSf4 6859
tests/datasets/test_datasets.py sha256=JLr7iYePnxIg7KE3Ide4FiPU6keXK8ZH2QxgCACkyLA 1727
tests/models/jax/test_jax_ite.py sha256=xOhyia8pv9vPhnd2EIBlRk4DFY2tgSPIMjw4sHTuHRs 1553
tests/models/jax/test_jax_model_utils.py sha256=SV8TJTJxfE7fN4PinK4TAampr6z_BkENpbxuskIPrD8 1117
tests/models/jax/test_jax_transformation_utils.py sha256=b9Fvp-Vxgyy4BMmRaHWFawS3KFCnk-ce96hNH0zamY0 1690
tests/models/torch/test_torch_flextenet.py sha256=f_4AK61nHbPNKAYwTm8S-pkqTat8eYwtaXqIYjJDWEc 3106
tests/models/torch/test_torch_pseudo_outcome_nets.py sha256=Ogi86Oa-WF3-O4wwwoFru3shJhQenL16pMeHqbqW2pc 3912
tests/models/torch/test_torch_representation_net.py sha256=R-DnOtEpC_WokjM3nLLeMPsdMmvyjhjtZGpQDcgnrvE 2656
tests/models/torch/test_torch_slearner.py sha256=8tJbGwIojVRmrC0uJzrqRZ43JsaJ1z_uI1u-fJBEl_8 5717
tests/models/torch/test_torch_snet.py sha256=JSYdaQo8cFVmvkqSIy-749hbT4fMk5HSHdVUdIrnnx0 4373
tests/models/torch/test_torch_tlearner.py sha256=En7B3BUuLFFH6o-kLpAOyVcxWWjMMn6i0JY4yC_y0cM 5218
catenets-0.2.3.dist-info/LICENSE sha256=h8iNCYQ6FyLlNf5AYVYpRwoBdowMjz3hlcz7Cs8AmTk 1520
catenets-0.2.3.dist-info/METADATA sha256=-qoH6q7pjQHvVGDpxhh9O9bRMmtv_IGYHLq36BuSjrY 7738
catenets-0.2.3.dist-info/WHEEL sha256=eB2aNjTYg-Xs2sC-30SiF6Qy3QhFjN9gbOG6r08MITc 108
catenets-0.2.3.dist-info/top_level.txt sha256=OAYTw1VO8n0ZWe0CRfrwsWDMBj9RXNj33rbNRXvpkHI 32
catenets-0.2.3.dist-info/RECORD

top_level.txt

catenets
docs
experiments
tests