Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cvtmle #91

Merged
merged 151 commits into from
Dec 6, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
151 commits
Select commit Hold shift + click to select a range
2d97b50
rename Parameter to Estimand
olivierlabayle Aug 1, 2023
d118649
rename parameter to estimand
olivierlabayle Aug 1, 2023
2d57de2
rename target to outcome
olivierlabayle Aug 2, 2023
20355f4
add scm
olivierlabayle Aug 2, 2023
ef44851
add show methods to scm and se
olivierlabayle Aug 3, 2023
96586f0
update equation struct to be fitted later
olivierlabayle Aug 3, 2023
37d3ec6
add se,scm fit!
olivierlabayle Aug 3, 2023
35fe73d
add fit estimands and reset scm
olivierlabayle Aug 3, 2023
79d3657
add another SCM constructor
olivierlabayle Aug 4, 2023
8e1749f
updated default Y model in scm
olivierlabayle Aug 4, 2023
4a7fb29
update utils
olivierlabayle Aug 4, 2023
6f090cd
update utils
olivierlabayle Aug 4, 2023
d0ba05c
update cache
olivierlabayle Aug 4, 2023
37e1486
more updates
olivierlabayle Aug 4, 2023
af22035
fix weighted fluctuation
olivierlabayle Aug 4, 2023
8dcfffa
fix some tests
olivierlabayle Aug 4, 2023
6afc9a9
fix more tests
olivierlabayle Aug 4, 2023
e447253
fix estimads tests
olivierlabayle Aug 4, 2023
157cedc
rename some files and add more tests
olivierlabayle Aug 6, 2023
b7f8f6f
update more tests
olivierlabayle Aug 6, 2023
9701cd7
tests working locally
olivierlabayle Aug 7, 2023
a6980cc
add more estimand checks
olivierlabayle Aug 7, 2023
5d0e9a2
more constructors
olivierlabayle Aug 7, 2023
a2064e9
update export list
olivierlabayle Aug 7, 2023
f801af2
add upstream methods
olivierlabayle Aug 7, 2023
eb180f7
add more generality for accepted graph
olivierlabayle Aug 7, 2023
3b6e05e
add more docstrings
olivierlabayle Aug 7, 2023
1deee66
rename some structs
olivierlabayle Aug 8, 2023
3da0049
add more docs
olivierlabayle Aug 9, 2023
24ca32e
update docs
olivierlabayle Aug 9, 2023
3342d49
add some constructors for estimands
olivierlabayle Aug 9, 2023
7428385
fix some names
olivierlabayle Aug 9, 2023
12be74c
more docs
olivierlabayle Aug 9, 2023
f666975
fix some constructors ambiguity
olivierlabayle Aug 10, 2023
19e27ee
update docs
olivierlabayle Aug 10, 2023
1ae36d4
update examples
olivierlabayle Aug 10, 2023
b956972
add extension
olivierlabayle Aug 10, 2023
8a315e5
more updates
olivierlabayle Aug 11, 2023
e4138e5
update docs and tests
olivierlabayle Aug 11, 2023
90e2d10
up deps and fix examples
olivierlabayle Aug 11, 2023
5fa1be4
update evotree dep
olivierlabayle Aug 13, 2023
e5cbefe
remove extension for now
olivierlabayle Aug 13, 2023
910bf04
actually remove ext
olivierlabayle Aug 13, 2023
7513f51
more tests
olivierlabayle Aug 13, 2023
32467ae
up def of with_encoder
olivierlabayle Aug 13, 2023
e0ed8f7
add data adaptive truncation
olivierlabayle Aug 14, 2023
564ecc5
rename plateau to truncate
olivierlabayle Aug 14, 2023
64c37d6
add dodgy workload for now
olivierlabayle Aug 16, 2023
654a21f
add fluctuation model
olivierlabayle Aug 20, 2023
4b8a8b3
some renames
olivierlabayle Aug 20, 2023
884bb9f
restructure project
olivierlabayle Aug 20, 2023
cef6d72
more refactorization
olivierlabayle Aug 20, 2023
c0ae075
fix ate double robustness test
olivierlabayle Aug 20, 2023
7e82c70
fix double robustness iate
olivierlabayle Aug 20, 2023
0ada1dc
add caching of covariate
olivierlabayle Aug 20, 2023
5596a08
cache training expected value
olivierlabayle Aug 20, 2023
ccc7fba
syntax adjustment
olivierlabayle Aug 20, 2023
b982cdd
add Statistics to test deps
olivierlabayle Aug 20, 2023
b518415
add precompilation again
olivierlabayle Aug 20, 2023
14487c7
add naive_plugin_estimate!
olivierlabayle Aug 20, 2023
7c36c4e
rename conditional mean to counterfactual mean
olivierlabayle Aug 20, 2023
1e174e4
add conditional distributions
olivierlabayle Aug 30, 2023
c9e62c2
add more content
olivierlabayle Sep 1, 2023
0c979e4
add work in progress
olivierlabayle Sep 5, 2023
f3cf47d
add more content
olivierlabayle Sep 6, 2023
d13962d
wip
olivierlabayle Sep 6, 2023
0ff82c8
add some tests
olivierlabayle Sep 12, 2023
3c16d83
add more work
olivierlabayle Sep 23, 2023
efdb459
fix estimands
olivierlabayle Sep 23, 2023
62b8657
clean testfile
olivierlabayle Sep 23, 2023
deaad87
more updates
olivierlabayle Sep 23, 2023
10ccef0
update gradient
olivierlabayle Sep 23, 2023
948e7b2
more updates
olivierlabayle Sep 23, 2023
9ce592d
fix dbl robust ate
olivierlabayle Sep 23, 2023
e57ff48
fix db robust iate
olivierlabayle Sep 23, 2023
8331876
fix non regression
olivierlabayle Sep 23, 2023
dceac15
fix more stuff
olivierlabayle Sep 23, 2023
2e99ee7
on my way to fixed tests
olivierlabayle Sep 24, 2023
d43e37b
fixed tests
olivierlabayle Sep 24, 2023
9afac05
add dr test for cv tmle and cv ose
olivierlabayle Sep 24, 2023
5391029
add type to method
olivierlabayle Sep 24, 2023
0721495
add types to structs fields
olivierlabayle Sep 24, 2023
5cafe37
clean some files
olivierlabayle Sep 25, 2023
1e77a44
towards inclusion of scm
olivierlabayle Sep 25, 2023
8a81c51
add scm and identification
olivierlabayle Sep 25, 2023
26b84c9
add compat entries
olivierlabayle Sep 25, 2023
2248e4a
remove optimize ordering as the cache allows storage of multiple esti…
olivierlabayle Sep 25, 2023
44ea3ab
only missing management to check
olivierlabayle Sep 25, 2023
414ce75
add estimand to TMLE/OSE estimates
olivierlabayle Sep 26, 2023
d9b9de8
add some tests for missing management
olivierlabayle Sep 26, 2023
35349f3
add some docstrings
olivierlabayle Oct 25, 2023
3007230
update docs
olivierlabayle Oct 26, 2023
02d42b7
fix some more docs
olivierlabayle Oct 30, 2023
1b8d4a9
update docs
olivierlabayle Oct 30, 2023
2e1a798
try see if docs are built
olivierlabayle Oct 30, 2023
d2e775d
update docs
olivierlabayle Oct 31, 2023
a761ea7
fix project
olivierlabayle Oct 31, 2023
2061cf6
add show method for scm
olivierlabayle Oct 31, 2023
29cea3e
fix code
olivierlabayle Oct 31, 2023
6e50645
fix code
olivierlabayle Oct 31, 2023
1677c91
add configurations back in
olivierlabayle Nov 1, 2023
b3170bc
add configuration
olivierlabayle Nov 6, 2023
78b94ec
update MLJBase dep and fix warnings
olivierlabayle Nov 6, 2023
71820a6
up AD dep
olivierlabayle Nov 6, 2023
ce918dd
up MLJ docs deps
olivierlabayle Nov 6, 2023
fac1bda
add some docs
olivierlabayle Nov 6, 2023
5fa052a
update docs
olivierlabayle Nov 6, 2023
25acbfd
add brute force ordering of parameters
olivierlabayle Nov 7, 2023
5d02722
add early stopping to brute force
olivierlabayle Nov 7, 2023
df25bf3
add brute_force and groups ordering of parameters
olivierlabayle Nov 9, 2023
dd6cc3a
change subtitle
olivierlabayle Nov 9, 2023
5d08999
add default models
olivierlabayle Nov 9, 2023
3cd90b5
add machine data caching option
olivierlabayle Nov 10, 2023
2a74417
add YAML extension
olivierlabayle Nov 13, 2023
082ceb8
add YAML to extra section for bacwards compatibility
olivierlabayle Nov 13, 2023
587f9af
limit config test to 1.9
olivierlabayle Nov 13, 2023
c44a993
add json serialization and std field to estimates
olivierlabayle Nov 17, 2023
0118f93
change some field names
olivierlabayle Nov 17, 2023
1236d15
add serialization of estimates
olivierlabayle Nov 17, 2023
c1373cf
fix 1.0 tests
olivierlabayle Nov 17, 2023
23d5393
drop nightly from CI
olivierlabayle Nov 17, 2023
5399c12
try explicitely close the file
olivierlabayle Nov 17, 2023
61ba7e0
try something else
olivierlabayle Nov 18, 2023
dc99bb3
fix bug in groups ordering
olivierlabayle Nov 18, 2023
3245613
try add + to read
olivierlabayle Nov 19, 2023
bda3fec
change read method's name
olivierlabayle Nov 20, 2023
d02633e
add FitFailedError
olivierlabayle Nov 21, 2023
a2cf0b9
add estimand to FitFailed exception
olivierlabayle Nov 21, 2023
ed9f899
enforce sorted tuples to make estimands equal
olivierlabayle Nov 22, 2023
28e3753
add ComposedEstimand and associated estimator
olivierlabayle Nov 22, 2023
6ed3afa
remove exports of some functions
olivierlabayle Nov 23, 2023
b49b6b4
add serialization of composite struct
olivierlabayle Nov 23, 2023
47873b8
fix serialization test
olivierlabayle Nov 23, 2023
e5b52da
change nuisance function counts function
olivierlabayle Nov 23, 2023
320f428
add ordering support for composed estimands
olivierlabayle Nov 24, 2023
3bc0c67
test emptyIC
olivierlabayle Nov 27, 2023
018bc47
remove bias from IC in OSE
olivierlabayle Nov 27, 2023
21349f5
add hotelling test
olivierlabayle Nov 28, 2023
897d8d4
test n_counts for composed
olivierlabayle Nov 28, 2023
245864a
add emptyIC for composed estimate
olivierlabayle Nov 28, 2023
f301b91
rename field std to cov
olivierlabayle Nov 28, 2023
34c1a06
relax some methods constraints
olivierlabayle Nov 28, 2023
3bd7796
add SJON serialization of ComposedEstimates
olivierlabayle Nov 29, 2023
62e16a7
fix 0 dim array returned sometimes and
olivierlabayle Nov 29, 2023
a67ab4d
fix non working sorting on tuples
olivierlabayle Dec 1, 2023
b4872f5
make config more flexible
olivierlabayle Dec 4, 2023
c2f143d
reexport OneSampleHotellingT2Test
olivierlabayle Dec 5, 2023
294f9fc
add kwargs to read_json and yaml to fix windows pb
olivierlabayle Dec 6, 2023
ba6f969
remove TMLE prefix in code
olivierlabayle Dec 6, 2023
b4083fb
remove precompilation workload for now
olivierlabayle Dec 6, 2023
9c05e87
up deps and remove unfortunate R file
olivierlabayle Dec 6, 2023
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@ jobs:
version:
- '1.6'
- '1'
- 'nightly'
os:
- ubuntu-latest
- macOS-latest
Expand Down
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,5 @@
build/
sandbox.jl
/Manifest.toml
/test/Manifest.toml
/test/Manifest.toml
/docs/Manifest.toml
35 changes: 26 additions & 9 deletions Project.toml
Original file line number Diff line number Diff line change
@@ -1,45 +1,62 @@
name = "TMLE"
uuid = "8afdd2fb-6e73-43df-8b62-b1650cd9c8cf"
authors = ["Olivier Labayle"]
version = "0.11.4"
version = "0.12.0"

[deps]
AbstractDifferentiation = "c29ec348-61ec-40c8-8164-b8c60e9d9f3d"
CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
Configurations = "5218b696-f38b-4ac9-8b61-a12ec717816d"
Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
GLM = "38e38edf-8417-5370-95a0-9cbb8c7f171a"
Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
HypothesisTests = "09f84164-cd44-5f33-b23f-e6b0d136a0d5"
JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
MLJGLMInterface = "caf8df21-4939-456d-ac9c-5fefbfb04c0c"
MLJModels = "d491faf4-2d78-11e9-2867-c94bc002c0b7"
MetaGraphsNext = "fa8bd995-216d-47f1-8a91-f3b68fbeb377"
Missings = "e1d29d7a-bbdc-5cf2-9ac0-f12de2c33e28"
PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
TableOperations = "ab02a1b2-a7df-11e8-156e-fb1833f50b87"
Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
YAML = "ddb6d928-2868-570f-bddf-ab3f9cf99eb6"
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

[weakdeps]
JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
YAML = "ddb6d928-2868-570f-bddf-ab3f9cf99eb6"

[extensions]
JSONExt = "JSON"
YAMLExt = "YAML"

[compat]
AbstractDifferentiation = "0.4, 0.5"
AbstractDifferentiation = "0.6.0"
CategoricalArrays = "0.10"
Configurations = "0.17"
Combinatorics = "1.0.2"
Distributions = "0.25"
GLM = "1.8.2"
HypothesisTests = "0.10"
Graphs = "1.8"
HypothesisTests = "0.10, 0.11"
JSON = "0.21.4"
LogExpFunctions = "0.3"
MLJBase = "0.19, 0.20, 0.21"
MLJBase = "1.0.1"
MLJGLMInterface = "0.3.4"
MLJModels = "0.15, 0.16"
MetaGraphsNext = "0.7"
Missings = "1.0"
PrecompileTools = "1.1.1"
PrettyTables = "2.2"
TableOperations = "1.2"
Tables = "1.6"
YAML = "0.4"
YAML = "0.4.9"
Zygote = "0.6"
PrecompileTools = "1.1.1"
julia = "1.6, 1.7, 1"

[extras]
JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
YAML = "ddb6d928-2868-570f-bddf-ab3f9cf99eb6"
5 changes: 2 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
![Codecov](https://img.shields.io/codecov/c/github/TARGENE/TMLE.jl/main)
![GitHub release (latest SemVer)](https://img.shields.io/github/v/release/TARGENE/TMLE.jl)

Targeted Minimum Loss-Based Estimation (TMLE) is a framework for efficient estimation of pathwise differentiable parameters. The goal of this package is to provide an implementation on top of the [MLJ](https://alan-turing-institute.github.io/MLJ.jl/dev/) framework.
This package enables the estimation of various Causal Inference related estimands using Targeted Minimum Loss-Based Estimation (TMLE). TMLE.jl is based on top of [MLJ](https://alan-turing-institute.github.io/MLJ.jl/dev/), which means any MLJ compliant machine-learning model can be used here.


**New to TMLE.jl?** The API design is still work in progress but you can already [get started now](https://targene.github.io/TMLE.jl/stable/)
**New to TMLE.jl?** [Get started now](https://targene.github.io/TMLE.jl/stable/)
Loading
Loading