├── .gitignore ├── LICENSE ├── README.md ├── dat └── pokec │ ├── README.md │ └── regional_subset │ ├── pokec_links.npz │ ├── pokec_links_processed.npz │ ├── pokec_links_svinet.txt │ └── profiles.pkl ├── poster └── poster.pdf └── src ├── .idea ├── misc.xml ├── modules.xml ├── src.iml ├── vcs.xml └── workspace.xml ├── relational_ERM ├── __init__.py ├── data_cleaning │ ├── __init__.py │ ├── pokec.py │ ├── run_simulate_baseline_neurips2019.sh │ ├── sbm_baseline_res.csv │ ├── simulate_baseline_sbm_neurips2019.py │ ├── simulate_treatment_outcome.py │ └── simulate_treatment_outcome_baseline.py ├── dataset │ ├── __init__.py │ └── dataset.py ├── effect_estimates │ ├── __init__.py │ ├── compute_ate.py │ ├── compute_single_ate.py │ ├── exogeneity_plotting.py │ └── process_baseline.py ├── graph_ops │ ├── __init__.py │ ├── representations.py │ ├── structure.py │ └── tensorflow.py ├── rerm_model │ ├── __init__.py │ ├── helpers.py │ ├── logging.py │ ├── run_classifier.py │ └── treatment_outcome_predictor.py ├── sampling │ ├── __init__.py │ ├── adapters.py │ ├── factories.py │ ├── negative_sampling.py │ └── sampling.h ├── submit_scripts │ ├── run_model.sh │ └── run_unsupervised.sh └── tensorflow_ops │ ├── __init__.py │ ├── adapter_ops.py │ ├── adjacency_to_edge_list.cpp │ ├── array_ops.py │ ├── batch_length_to_segment.cpp │ ├── biased_walk_dataset.cpp │ ├── concatenate_slices.cpp │ ├── dataset_ops.py │ ├── induced_ego_sample.cpp │ ├── induced_subgraph.cpp │ ├── p_sampling_dataset.cpp │ ├── packed_to_sparse.cpp │ ├── random_walk_dataset.cpp │ ├── repeat.cpp │ └── uniform_edge_dataset.cpp ├── rerm.yml ├── semi_parametric_estimation ├── __init__.py ├── ate.py ├── att.py └── helpers.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/README.md -------------------------------------------------------------------------------- /dat/pokec/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/dat/pokec/README.md -------------------------------------------------------------------------------- /dat/pokec/regional_subset/pokec_links.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/dat/pokec/regional_subset/pokec_links.npz -------------------------------------------------------------------------------- /dat/pokec/regional_subset/pokec_links_processed.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/dat/pokec/regional_subset/pokec_links_processed.npz -------------------------------------------------------------------------------- /dat/pokec/regional_subset/pokec_links_svinet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/dat/pokec/regional_subset/pokec_links_svinet.txt -------------------------------------------------------------------------------- /dat/pokec/regional_subset/profiles.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/dat/pokec/regional_subset/profiles.pkl -------------------------------------------------------------------------------- /poster/poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/poster/poster.pdf -------------------------------------------------------------------------------- /src/.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/.idea/misc.xml -------------------------------------------------------------------------------- /src/.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/.idea/modules.xml -------------------------------------------------------------------------------- /src/.idea/src.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/.idea/src.iml -------------------------------------------------------------------------------- /src/.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/.idea/vcs.xml -------------------------------------------------------------------------------- /src/.idea/workspace.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/.idea/workspace.xml -------------------------------------------------------------------------------- /src/relational_ERM/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/pokec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/data_cleaning/pokec.py -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/run_simulate_baseline_neurips2019.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/data_cleaning/run_simulate_baseline_neurips2019.sh -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/sbm_baseline_res.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/data_cleaning/sbm_baseline_res.csv -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/simulate_baseline_sbm_neurips2019.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/data_cleaning/simulate_baseline_sbm_neurips2019.py -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/simulate_treatment_outcome.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/data_cleaning/simulate_treatment_outcome.py -------------------------------------------------------------------------------- /src/relational_ERM/data_cleaning/simulate_treatment_outcome_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/data_cleaning/simulate_treatment_outcome_baseline.py -------------------------------------------------------------------------------- /src/relational_ERM/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/dataset/dataset.py -------------------------------------------------------------------------------- /src/relational_ERM/effect_estimates/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/effect_estimates/compute_ate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/effect_estimates/compute_ate.py -------------------------------------------------------------------------------- /src/relational_ERM/effect_estimates/compute_single_ate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/effect_estimates/compute_single_ate.py -------------------------------------------------------------------------------- /src/relational_ERM/effect_estimates/exogeneity_plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/effect_estimates/exogeneity_plotting.py -------------------------------------------------------------------------------- /src/relational_ERM/effect_estimates/process_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/effect_estimates/process_baseline.py -------------------------------------------------------------------------------- /src/relational_ERM/graph_ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/graph_ops/representations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/graph_ops/representations.py -------------------------------------------------------------------------------- /src/relational_ERM/graph_ops/structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/graph_ops/structure.py -------------------------------------------------------------------------------- /src/relational_ERM/graph_ops/tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/graph_ops/tensorflow.py -------------------------------------------------------------------------------- /src/relational_ERM/rerm_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/rerm_model/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/rerm_model/helpers.py -------------------------------------------------------------------------------- /src/relational_ERM/rerm_model/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/rerm_model/logging.py -------------------------------------------------------------------------------- /src/relational_ERM/rerm_model/run_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/rerm_model/run_classifier.py -------------------------------------------------------------------------------- /src/relational_ERM/rerm_model/treatment_outcome_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/rerm_model/treatment_outcome_predictor.py -------------------------------------------------------------------------------- /src/relational_ERM/sampling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/relational_ERM/sampling/adapters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/sampling/adapters.py -------------------------------------------------------------------------------- /src/relational_ERM/sampling/factories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/sampling/factories.py -------------------------------------------------------------------------------- /src/relational_ERM/sampling/negative_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/sampling/negative_sampling.py -------------------------------------------------------------------------------- /src/relational_ERM/sampling/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/sampling/sampling.h -------------------------------------------------------------------------------- /src/relational_ERM/submit_scripts/run_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/submit_scripts/run_model.sh -------------------------------------------------------------------------------- /src/relational_ERM/submit_scripts/run_unsupervised.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/submit_scripts/run_unsupervised.sh -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/__init__.py -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/adapter_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/adapter_ops.py -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/adjacency_to_edge_list.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/adjacency_to_edge_list.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/array_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/array_ops.py -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/batch_length_to_segment.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/batch_length_to_segment.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/biased_walk_dataset.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/biased_walk_dataset.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/concatenate_slices.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/concatenate_slices.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/dataset_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/dataset_ops.py -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/induced_ego_sample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/induced_ego_sample.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/induced_subgraph.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/induced_subgraph.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/p_sampling_dataset.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/p_sampling_dataset.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/packed_to_sparse.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/packed_to_sparse.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/random_walk_dataset.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/random_walk_dataset.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/repeat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/repeat.cpp -------------------------------------------------------------------------------- /src/relational_ERM/tensorflow_ops/uniform_edge_dataset.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/relational_ERM/tensorflow_ops/uniform_edge_dataset.cpp -------------------------------------------------------------------------------- /src/rerm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/rerm.yml -------------------------------------------------------------------------------- /src/semi_parametric_estimation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/semi_parametric_estimation/ate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/semi_parametric_estimation/ate.py -------------------------------------------------------------------------------- /src/semi_parametric_estimation/att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/semi_parametric_estimation/att.py -------------------------------------------------------------------------------- /src/semi_parametric_estimation/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/semi_parametric_estimation/helpers.py -------------------------------------------------------------------------------- /src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vveitch/causal-network-embeddings/HEAD/src/setup.py --------------------------------------------------------------------------------