├── .devcontainer ├── .p10k.zsh └── devcontainer.json ├── .github └── workflows │ ├── ci-build.yml │ ├── codeql-analysis.yml │ ├── pypi-publish.yml │ └── sphinx-build.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── Dockerfile ├── LICENSE ├── README.md ├── SECURITY.md ├── SUPPORT.md ├── docs ├── Makefile ├── README.md └── src │ ├── conf.py │ ├── index.md │ └── templates │ └── autoapi │ └── python │ └── module.rst ├── examples ├── csuite_example.ipynb ├── lightning_example │ ├── Readme.md │ ├── config │ │ ├── data.yaml │ │ ├── model.yaml │ │ └── training.yaml │ ├── data_module.py │ ├── lightning_module.py │ └── model_analysis.ipynb ├── multi_investment_sales_attribution.ipynb └── rhino_ecoli_example.ipynb ├── poetry.lock ├── pyproject.toml ├── research_experiments ├── avid │ ├── README.md │ ├── images │ │ ├── coinrun1.gif │ │ ├── coinrun2.gif │ │ ├── coinrun3.gif │ │ ├── coinrun4.gif │ │ ├── coinrun5.gif │ │ ├── coinrun6.gif │ │ ├── coinrun7.gif │ │ ├── rt1_1.gif │ │ ├── rt1_2.gif │ │ ├── rt1_3.gif │ │ ├── rt1_4.gif │ │ └── rt1_5.gif │ ├── latent_diffusion │ │ ├── README.md │ │ ├── configs │ │ │ ├── eval │ │ │ │ ├── act_cond_diffusion_145M.yaml │ │ │ │ ├── avid_145M.yaml │ │ │ │ ├── control_net_170M.yaml │ │ │ │ ├── control_net_full.yaml │ │ │ │ ├── dynamicrafter_pretrained.yaml │ │ │ │ ├── full_finetune.yaml │ │ │ │ └── poe_145M.yaml │ │ │ └── train │ │ │ │ ├── act_cond_diffusion_11M.yaml │ │ │ │ ├── act_cond_diffusion_145M.yaml │ │ │ │ ├── act_cond_diffusion_34M.yaml │ │ │ │ ├── avid │ │ │ │ ├── avid_11M.yaml │ │ │ │ ├── avid_145M.yaml │ │ │ │ └── avid_34M.yaml │ │ │ │ ├── classifier │ │ │ │ ├── categorical_classifier_for_guidance.yaml │ │ │ │ └── mse_predictor_for_metrics.yaml │ │ │ │ ├── control_net │ │ │ │ ├── act_control.yaml │ │ │ │ ├── act_control_lite_10M.yaml │ │ │ │ ├── act_control_lite_170M.yaml │ │ │ │ ├── act_control_lite_38M.yaml │ │ │ │ ├── control_net.yaml │ │ │ │ ├── control_net_lite_10M.yaml │ │ │ │ ├── control_net_lite_170M.yaml │ │ │ │ ├── control_net_lite_38M.yaml │ │ │ │ └── dynamicrafter_512.yaml │ │ │ │ ├── dynamicrafter_512.yaml │ │ │ │ └── dynamicrafter_512_action_finetune.yaml │ │ ├── libs │ │ │ ├── dynamicrafter │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── lvdm │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── basics.py │ │ │ │ │ ├── common.py │ │ │ │ │ ├── distributions.py │ │ │ │ │ ├── ema.py │ │ │ │ │ ├── models │ │ │ │ │ │ ├── action_predictor.py │ │ │ │ │ │ ├── autoencoder.py │ │ │ │ │ │ ├── ddpm3d.py │ │ │ │ │ │ ├── samplers │ │ │ │ │ │ │ ├── ddim.py │ │ │ │ │ │ │ └── ddim_multiplecond.py │ │ │ │ │ │ └── utils_diffusion.py │ │ │ │ │ ├── modules │ │ │ │ │ │ ├── attention.py │ │ │ │ │ │ ├── encoders │ │ │ │ │ │ │ ├── condition.py │ │ │ │ │ │ │ └── resampler.py │ │ │ │ │ │ ├── networks │ │ │ │ │ │ │ ├── ae_modules.py │ │ │ │ │ │ │ ├── control_net.py │ │ │ │ │ │ │ └── openaimodel3d.py │ │ │ │ │ │ └── x_transformer.py │ │ │ │ │ └── utils │ │ │ │ │ │ ├── callbacks.py │ │ │ │ │ │ ├── data.py │ │ │ │ │ │ ├── eval.py │ │ │ │ │ │ ├── save_video.py │ │ │ │ │ │ ├── train.py │ │ │ │ │ │ └── utils.py │ │ │ │ └── setup.py │ │ │ └── octo │ │ │ │ ├── HEAD │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── octo │ │ │ │ ├── __init__.py │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── dataset.py │ │ │ │ │ ├── obs_transforms.py │ │ │ │ │ ├── oxe │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── oxe_dataset_configs.py │ │ │ │ │ │ ├── oxe_dataset_mixes.py │ │ │ │ │ │ └── oxe_standardization_transforms.py │ │ │ │ │ ├── traj_transforms.py │ │ │ │ │ └── utils │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── data_utils.py │ │ │ │ │ │ ├── goal_relabeling.py │ │ │ │ │ │ ├── task_augmentation.py │ │ │ │ │ │ └── text_processing.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── spec.py │ │ │ │ └── setup.py │ │ ├── poetry.lock │ │ ├── pyproject.toml │ │ ├── scripts │ │ │ ├── eval │ │ │ │ ├── eval_avid.py │ │ │ │ ├── eval_controlnet.py │ │ │ │ ├── eval_diffusion.py │ │ │ │ └── eval_mixed_diffusion.py │ │ │ ├── train.sh │ │ │ ├── train_avid.py │ │ │ ├── train_classifier.py │ │ │ ├── train_control_net.py │ │ │ └── train_diffusion.py │ │ └── src │ │ │ └── ldwma │ │ │ ├── __init__.py │ │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── droid.py │ │ │ ├── rtx.py │ │ │ └── utils.py │ │ │ ├── lightning │ │ │ └── data_modules │ │ │ │ └── rtx.py │ │ │ └── models │ │ │ ├── avid.py │ │ │ ├── control_net.py │ │ │ └── wrappers │ │ │ ├── classifier_guidance.py │ │ │ └── mixed_diffusion.py │ ├── libs │ │ └── avid_utils │ │ │ ├── __init__.py │ │ │ ├── avid_utils │ │ │ ├── __init__.py │ │ │ ├── eval.py │ │ │ ├── helpers.py │ │ │ ├── image.py │ │ │ ├── metrics.py │ │ │ ├── normalization.py │ │ │ └── video.py │ │ │ └── setup.py │ └── pixel_diffusion │ │ ├── README.md │ │ ├── config │ │ ├── act_cond_model_22M_coinrun_500lvl.yaml │ │ ├── act_cond_model_22M_coinrun_500lvl_cfg.yaml │ │ ├── act_cond_model_71M_coinrun_500lvl.yaml │ │ ├── act_cond_model_71M_coinrun_500lvl_cfg.yaml │ │ ├── avid │ │ │ ├── avid_22M_coinrun_500lvl.yaml │ │ │ ├── avid_71M_coinrun_500lvl.yaml │ │ │ └── base_avid_config.yaml │ │ ├── base_video_config.yaml │ │ ├── classifier │ │ │ ├── base_classifier_config.yaml │ │ │ ├── classifier_for_eval.yaml │ │ │ └── classifier_for_guidance_500lvl.yaml │ │ ├── control_net │ │ │ ├── base_controlnet_config.yaml │ │ │ ├── controlnet_22M_coinrun_500lvl.yaml │ │ │ └── controlnet_full_coinrun_500lvl.yaml │ │ ├── eval │ │ │ ├── act_cond_model_22M_coinrun_500lvl.yaml │ │ │ ├── act_cond_model_71M_coinrun_500lvl.yaml │ │ │ ├── avid_22M_coinrun_500lvl.yaml │ │ │ ├── avid_71M_coinrun_500lvl.yaml │ │ │ ├── controlnet_22M_coinrun_500lvl.yaml │ │ │ ├── controlnet_full_coinrun_500lvl.yaml │ │ │ ├── full_finetune_coinrun_500lvl.yaml │ │ │ └── pretrained_model.yaml │ │ ├── full_finetune_with_act_500lvl.yaml │ │ └── pretrained_model.yaml │ │ ├── poetry.lock │ │ ├── pyproject.toml │ │ ├── scripts │ │ ├── datasets │ │ │ ├── sample_coinrun_datasets.sh │ │ │ ├── sample_procgen_data.py │ │ │ └── sample_procgen_dataset.sh │ │ ├── eval │ │ │ ├── eval_avid.py │ │ │ ├── eval_controlnet.py │ │ │ └── eval_diffusion.py │ │ ├── train_avid.py │ │ ├── train_classifier.py │ │ ├── train_control_net.py │ │ └── train_diffusion.py │ │ └── src │ │ └── dwma │ │ ├── __init__.py │ │ ├── datasets │ │ ├── __init__.py │ │ ├── file_dataset.py │ │ ├── procgen_dataset.py │ │ └── procgen_sampler.py │ │ ├── lightning │ │ ├── cli │ │ │ └── cli.py │ │ ├── data_modules │ │ │ └── procgen_data.py │ │ └── modules │ │ │ ├── classifier_module.py │ │ │ └── diffusion_module.py │ │ ├── models │ │ ├── __init__.py │ │ ├── avid.py │ │ ├── classifier.py │ │ ├── classifier_guided_diffusion.py │ │ ├── control_net.py │ │ ├── diffusion_control_net.py │ │ ├── mixed_diffusion.py │ │ └── video_diffusion_pytorch │ │ │ ├── LICENSE │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── blocks.py │ │ │ ├── diffusion.py │ │ │ └── unet3d.py │ │ └── setup.py ├── cond_fip │ ├── README.md │ ├── poetry.lock │ ├── pyproject.toml │ └── src │ │ └── cond_fip │ │ ├── config │ │ ├── cond_fip_inference.yaml │ │ ├── cond_fip_training.yaml │ │ ├── encoder_training.yaml │ │ ├── numpy_tensor_data_module.yaml │ │ └── synthetic_data_module.yaml │ │ ├── entrypoint_test.py │ │ ├── launchers │ │ ├── inference_cond_fip.py │ │ ├── train_cond_fip.py │ │ └── train_encoder.py │ │ ├── models │ │ ├── amortized_models.py │ │ └── decoder_layer.py │ │ └── tasks │ │ ├── cond_fip_inference.py │ │ ├── cond_fip_training.py │ │ └── encoder_training.py ├── fip │ ├── README.md │ ├── poetry.lock │ ├── pyproject.toml │ └── src │ │ └── fip │ │ ├── __init__.py │ │ ├── config │ │ ├── amortization │ │ │ └── leaf_prediction.yaml │ │ ├── numpy_tensor_data_module.yaml │ │ ├── scm_learning_with_ground_truth │ │ │ ├── scm_learning_true_graph.yaml │ │ │ └── scm_learning_true_perm.yaml │ │ ├── scm_learning_with_predicted_truth │ │ │ └── scm_learning_predicted_leaf.yaml │ │ └── synthetic_data_module.yaml │ │ ├── data_generation │ │ ├── README.md │ │ ├── avici_data.py │ │ ├── avici_data.sh │ │ ├── config_data.py │ │ ├── csuite_data.py │ │ ├── csuite_data.sh │ │ ├── normalizing_flow_data.py │ │ ├── normalizing_flow_data.sh │ │ └── sem_factory.py │ │ ├── data_modules │ │ ├── numpy_tensor_data_module.py │ │ └── synthetic_data_module.py │ │ ├── entrypoint.py │ │ ├── launchers │ │ ├── README.md │ │ ├── amortization.py │ │ ├── basic_commands.py │ │ ├── scm_learning_with_ground_truth.py │ │ └── scm_learning_with_predicted_truth.py │ │ ├── methods │ │ ├── base_method_module.py │ │ ├── jacobian.py │ │ └── scm_learning.py │ │ ├── models │ │ ├── amortized_models.py │ │ └── causal_models.py │ │ ├── my_cli.py │ │ ├── task_utils │ │ ├── leaf_functions.py │ │ └── learnable_loss.py │ │ └── tasks │ │ ├── amortization │ │ └── leaf_prediction.py │ │ ├── scm_learning_with_ground_truth │ │ ├── scm_learning_true_graph.py │ │ └── scm_learning_true_perm.py │ │ └── scm_learning_with_predicted_truth │ │ └── scm_learning_predicted_leaf.py └── scotch │ ├── README.md │ ├── poetry.lock │ ├── pyproject.toml │ └── src │ └── scotch │ ├── dataset_generation │ ├── example_sdes │ │ ├── lorenz_sde.py │ │ └── yeast_glycolysis.py │ ├── generate_and_save_data.py │ └── generate_trajectories.py │ ├── experiments │ ├── dream3.py │ ├── lorenz.py │ ├── netsim.py │ └── yeast.py │ ├── latent_learning │ ├── functional_relationship.py │ ├── graph_distribution.py │ ├── scotch_data_module.py │ ├── scotch_module.py │ └── scotch_nns.py │ ├── scotch_utils │ ├── graph_metrics.py │ └── scotch_utils.py │ └── sdes │ ├── __init__.py │ ├── scotch_sdes.py │ ├── sde_modules.py │ └── sdes_core.py ├── src └── causica │ ├── __init__.py │ ├── config │ └── lightning │ │ ├── default_data.yaml │ │ ├── default_gaussian.yaml │ │ └── default_spline.yaml │ ├── data_generation │ ├── __init__.py │ ├── generate_data.py │ └── samplers │ │ ├── __init__.py │ │ ├── functional_relationships_sampler.py │ │ ├── noise_dist_sampler.py │ │ ├── sampler.py │ │ └── sem_sampler.py │ ├── datasets │ ├── __init__.py │ ├── causal_dataset.py │ ├── causica_dataset_format │ │ ├── __init__.py │ │ ├── load.py │ │ └── save.py │ ├── interventional_data.py │ ├── loaded_expert_graph_container.py │ ├── normalization.py │ ├── samplers.py │ ├── synthetic_dataset.py │ ├── tensordict_utils.py │ ├── timeseries_dataset.py │ └── variable_types.py │ ├── distributions │ ├── __init__.py │ ├── adjacency │ │ ├── __init__.py │ │ ├── adjacency_distributions.py │ │ ├── constrained_adjacency_distributions.py │ │ ├── edges_per_node_erdos_renyi.py │ │ ├── enco.py │ │ ├── erdos_renyi.py │ │ ├── fixed_adjacency_distribution.py │ │ ├── geometric_random_graph.py │ │ ├── gibbs_dag_prior.py │ │ ├── scale_free.py │ │ ├── stochastic_block_model.py │ │ ├── temporal_adjacency_distributions.py │ │ ├── three_way.py │ │ └── watts_strogatz.py │ ├── distribution_module.py │ ├── gumbel_binary.py │ ├── noise │ │ ├── __init__.py │ │ ├── bernoulli.py │ │ ├── categorical.py │ │ ├── joint.py │ │ ├── noise.py │ │ ├── spline │ │ │ ├── __init__.py │ │ │ ├── bayesiains_nsf_rqs.py │ │ │ ├── rational_quadratic_transform.py │ │ │ └── spline.py │ │ ├── univariate_cauchy.py │ │ ├── univariate_laplace.py │ │ └── univariate_normal.py │ ├── signed_uniform.py │ └── transforms │ │ ├── __init__.py │ │ ├── base.py │ │ ├── joint.py │ │ └── tensor_to_tensordict.py │ ├── functional_relationships │ ├── __init__.py │ ├── deci_functional_relationships.py │ ├── do_functional_relationships.py │ ├── functional_relationships.py │ ├── heteroscedastic_rff_functional_relationships.py │ ├── linear_functional_relationships.py │ ├── rff_functional_relationships.py │ └── temporal_functional_relationships.py │ ├── graph │ ├── __init__.py │ ├── dag_constraint.py │ └── evaluation_metrics.py │ ├── lightning │ ├── __init__.py │ ├── callbacks.py │ ├── cli.py │ ├── data_modules │ │ ├── __init__.py │ │ ├── basic_data_module.py │ │ ├── deci_data_module.py │ │ ├── synthetic_data_module.py │ │ └── variable_spec_data.py │ ├── loggers.py │ ├── main.py │ └── modules │ │ ├── __init__.py │ │ ├── deci_module.py │ │ └── variable_spec_module.py │ ├── mlflow_helpers.py │ ├── nn │ ├── __init__.py │ ├── deci_embed_nn.py │ ├── fip │ │ ├── __init__.py │ │ ├── attention_layer.py │ │ ├── avici_encoder.py │ │ ├── decoder_layer.py │ │ └── embeddings.py │ └── temporal_embed_nn.py │ ├── py.typed │ ├── sem │ ├── __init__.py │ ├── distribution_parameters_sem.py │ ├── sem_distribution.py │ ├── structural_equation_model.py │ └── temporal_distribution_parameters_sem.py │ ├── training │ ├── __init__.py │ ├── auglag.py │ ├── evaluation.py │ └── per_variable_metrics.py │ └── triangular_transformations.py └── test ├── __init__.py ├── data_generation └── test_generate_data.py ├── datasets ├── test_datasets.py ├── test_loaded_expert_graph_container.py ├── test_samplers.py ├── test_standardizer.py ├── test_synthetic_dataset.py ├── test_tensordict_utils.py └── test_timeseries_dataset.py ├── distributions ├── __init__.py ├── adjacency │ ├── test_adjacency_distributions.py │ ├── test_constrained_adjacency_distributions.py │ ├── test_edges_per_node_erdos_renyi.py │ ├── test_enco.py │ ├── test_erdos_renyi.py │ ├── test_geometric_random_graph.py │ ├── test_gibbs_dag_prior.py │ ├── test_scale_free.py │ ├── test_stochastic_block_model.py │ ├── test_three_way.py │ └── test_watts_strogatz.py ├── noise │ ├── __init__.py │ ├── spline │ │ ├── __init__.py │ │ └── test_splines.py │ ├── test_bernoulli.py │ ├── test_categorical.py │ ├── test_joint.py │ ├── test_univariate_cauchy.py │ ├── test_univariate_laplace.py │ └── test_univariate_normal.py ├── test_sem_distribution.py └── transforms │ ├── test_base.py │ ├── test_joint.py │ └── test_transform_modules.py ├── functional_relationships ├── __init__.py ├── test_do_functional_relationships.py └── test_functional_relationships.py ├── integration ├── __init__.py ├── decimodule.pt ├── test_evaluation.py ├── test_pytorch_lightning.py ├── test_save_load_csuite.py └── test_timeseries_dataset.py ├── lightning ├── __init__.py ├── test_basic_data_module.py ├── test_callbacks.py ├── test_loggers.py └── test_variable_spec_data.py ├── metrics ├── __init__.py └── test_graph_eval_metrics.py ├── nn ├── test_deci_embed_nn.py └── test_temporal_embed_nn.py ├── sem ├── __init__.py ├── test_sem.py ├── test_temporal_distribution_parameters_sem.py └── test_treatment_effects.py ├── test_dag_constraint.py ├── test_triangular_transformations.py └── training ├── __init__.py └── test_auglag.py /.devcontainer/.p10k.zsh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.devcontainer/.p10k.zsh -------------------------------------------------------------------------------- /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.github/workflows/ci-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.github/workflows/ci-build.yml -------------------------------------------------------------------------------- /.github/workflows/codeql-analysis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.github/workflows/codeql-analysis.yml -------------------------------------------------------------------------------- /.github/workflows/pypi-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.github/workflows/pypi-publish.yml -------------------------------------------------------------------------------- /.github/workflows/sphinx-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.github/workflows/sphinx-build.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/SECURITY.md -------------------------------------------------------------------------------- /SUPPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/SUPPORT.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/src/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/docs/src/conf.py -------------------------------------------------------------------------------- /docs/src/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/docs/src/index.md -------------------------------------------------------------------------------- /docs/src/templates/autoapi/python/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/docs/src/templates/autoapi/python/module.rst -------------------------------------------------------------------------------- /examples/csuite_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/csuite_example.ipynb -------------------------------------------------------------------------------- /examples/lightning_example/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/Readme.md -------------------------------------------------------------------------------- /examples/lightning_example/config/data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/config/data.yaml -------------------------------------------------------------------------------- /examples/lightning_example/config/model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/config/model.yaml -------------------------------------------------------------------------------- /examples/lightning_example/config/training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/config/training.yaml -------------------------------------------------------------------------------- /examples/lightning_example/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/data_module.py -------------------------------------------------------------------------------- /examples/lightning_example/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/lightning_module.py -------------------------------------------------------------------------------- /examples/lightning_example/model_analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/lightning_example/model_analysis.ipynb -------------------------------------------------------------------------------- /examples/multi_investment_sales_attribution.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/multi_investment_sales_attribution.ipynb -------------------------------------------------------------------------------- /examples/rhino_ecoli_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/examples/rhino_ecoli_example.ipynb -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/pyproject.toml -------------------------------------------------------------------------------- /research_experiments/avid/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/README.md -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun1.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun2.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun3.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun4.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun4.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun5.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun5.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun6.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun6.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/coinrun7.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/coinrun7.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/rt1_1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/rt1_1.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/rt1_2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/rt1_2.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/rt1_3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/rt1_3.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/rt1_4.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/rt1_4.gif -------------------------------------------------------------------------------- /research_experiments/avid/images/rt1_5.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/images/rt1_5.gif -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/README.md -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/act_cond_diffusion_145M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/act_cond_diffusion_145M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/avid_145M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/avid_145M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/control_net_170M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/control_net_170M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/control_net_full.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/control_net_full.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/dynamicrafter_pretrained.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/dynamicrafter_pretrained.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/full_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/full_finetune.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/eval/poe_145M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/eval/poe_145M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/act_cond_diffusion_11M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/act_cond_diffusion_11M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/act_cond_diffusion_145M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/act_cond_diffusion_145M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/act_cond_diffusion_34M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/act_cond_diffusion_34M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/avid/avid_11M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/avid/avid_11M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/avid/avid_145M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/avid/avid_145M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/avid/avid_34M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/avid/avid_34M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/classifier/categorical_classifier_for_guidance.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/classifier/categorical_classifier_for_guidance.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/classifier/mse_predictor_for_metrics.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/classifier/mse_predictor_for_metrics.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/act_control.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/act_control.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/act_control_lite_10M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/act_control_lite_10M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/act_control_lite_170M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/act_control_lite_170M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/act_control_lite_38M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/act_control_lite_38M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/control_net.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/control_net.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/control_net_lite_10M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/control_net_lite_10M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/control_net_lite_170M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/control_net_lite_170M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/control_net_lite_38M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/control_net_lite_38M.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/control_net/dynamicrafter_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/control_net/dynamicrafter_512.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/dynamicrafter_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/dynamicrafter_512.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/configs/train/dynamicrafter_512_action_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/configs/train/dynamicrafter_512_action_finetune.yaml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/LICENSE -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/README.md -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/basics.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/common.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/distributions.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/ema.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/action_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/action_predictor.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/autoencoder.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/ddpm3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/ddpm3d.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/samplers/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/samplers/ddim.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/samplers/ddim_multiplecond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/samplers/ddim_multiplecond.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/utils_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/models/utils_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/attention.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/encoders/condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/encoders/condition.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/encoders/resampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/encoders/resampler.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/networks/ae_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/networks/ae_modules.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/networks/control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/networks/control_net.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/networks/openaimodel3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/networks/openaimodel3d.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/modules/x_transformer.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/callbacks.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/data.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/eval.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/save_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/save_video.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/train.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/lvdm/utils/utils.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/dynamicrafter/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/dynamicrafter/setup.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/HEAD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/HEAD -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/LICENSE -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/README.md -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/dataset.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/obs_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/obs_transforms.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/__init__.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/oxe_dataset_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/oxe_dataset_configs.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/oxe_dataset_mixes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/oxe_dataset_mixes.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/oxe_standardization_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/oxe/oxe_standardization_transforms.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/traj_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/traj_transforms.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/data_utils.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/goal_relabeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/goal_relabeling.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/task_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/task_augmentation.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/text_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/data/utils/text_processing.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/octo/utils/spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/octo/utils/spec.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/libs/octo/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/libs/octo/setup.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/poetry.lock -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/pyproject.toml -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/eval/eval_avid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/eval/eval_avid.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/eval/eval_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/eval/eval_controlnet.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/eval/eval_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/eval/eval_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/eval/eval_mixed_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/eval/eval_mixed_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/train.sh -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/train_avid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/train_avid.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/train_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/train_classifier.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/train_control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/train_control_net.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/scripts/train_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/scripts/train_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/datasets/droid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/datasets/droid.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/datasets/rtx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/datasets/rtx.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/datasets/utils.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/lightning/data_modules/rtx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/lightning/data_modules/rtx.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/models/avid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/models/avid.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/models/control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/models/control_net.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/models/wrappers/classifier_guidance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/models/wrappers/classifier_guidance.py -------------------------------------------------------------------------------- /research_experiments/avid/latent_diffusion/src/ldwma/models/wrappers/mixed_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/latent_diffusion/src/ldwma/models/wrappers/mixed_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/avid_utils/eval.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/avid_utils/helpers.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/avid_utils/image.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/avid_utils/metrics.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/avid_utils/normalization.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/avid_utils/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/avid_utils/video.py -------------------------------------------------------------------------------- /research_experiments/avid/libs/avid_utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/libs/avid_utils/setup.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/README.md -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/act_cond_model_22M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/act_cond_model_22M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/act_cond_model_22M_coinrun_500lvl_cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/act_cond_model_22M_coinrun_500lvl_cfg.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/act_cond_model_71M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/act_cond_model_71M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/act_cond_model_71M_coinrun_500lvl_cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/act_cond_model_71M_coinrun_500lvl_cfg.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/avid/avid_22M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/avid/avid_22M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/avid/avid_71M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/avid/avid_71M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/avid/base_avid_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/avid/base_avid_config.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/base_video_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/base_video_config.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/classifier/base_classifier_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/classifier/base_classifier_config.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/classifier/classifier_for_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/classifier/classifier_for_eval.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/classifier/classifier_for_guidance_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/classifier/classifier_for_guidance_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/control_net/base_controlnet_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/control_net/base_controlnet_config.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/control_net/controlnet_22M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/control_net/controlnet_22M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/control_net/controlnet_full_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/control_net/controlnet_full_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/act_cond_model_22M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/act_cond_model_22M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/act_cond_model_71M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/act_cond_model_71M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/avid_22M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/avid_22M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/avid_71M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/avid_71M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/controlnet_22M_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/controlnet_22M_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/controlnet_full_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/controlnet_full_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/full_finetune_coinrun_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/full_finetune_coinrun_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/eval/pretrained_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/eval/pretrained_model.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/full_finetune_with_act_500lvl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/full_finetune_with_act_500lvl.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/config/pretrained_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/config/pretrained_model.yaml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/poetry.lock -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/pyproject.toml -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/datasets/sample_coinrun_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/datasets/sample_coinrun_datasets.sh -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/datasets/sample_procgen_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/datasets/sample_procgen_data.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/datasets/sample_procgen_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/datasets/sample_procgen_dataset.sh -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/eval/eval_avid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/eval/eval_avid.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/eval/eval_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/eval/eval_controlnet.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/eval/eval_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/eval/eval_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/train_avid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/train_avid.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/train_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/train_classifier.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/train_control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/train_control_net.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/scripts/train_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/scripts/train_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/datasets/file_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/datasets/file_dataset.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/datasets/procgen_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/datasets/procgen_dataset.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/datasets/procgen_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/datasets/procgen_sampler.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/lightning/cli/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/lightning/cli/cli.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/lightning/data_modules/procgen_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/lightning/data_modules/procgen_data.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/lightning/modules/classifier_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/lightning/modules/classifier_module.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/lightning/modules/diffusion_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/lightning/modules/diffusion_module.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/avid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/avid.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/classifier.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/classifier_guided_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/classifier_guided_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/control_net.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/diffusion_control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/diffusion_control_net.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/mixed_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/mixed_diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/LICENSE -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/attention.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/blocks.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/diffusion.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/unet3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/models/video_diffusion_pytorch/unet3d.py -------------------------------------------------------------------------------- /research_experiments/avid/pixel_diffusion/src/dwma/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/avid/pixel_diffusion/src/dwma/setup.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/README.md -------------------------------------------------------------------------------- /research_experiments/cond_fip/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/poetry.lock -------------------------------------------------------------------------------- /research_experiments/cond_fip/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/pyproject.toml -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/config/cond_fip_inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/config/cond_fip_inference.yaml -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/config/cond_fip_training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/config/cond_fip_training.yaml -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/config/encoder_training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/config/encoder_training.yaml -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/config/numpy_tensor_data_module.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/config/numpy_tensor_data_module.yaml -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/config/synthetic_data_module.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/config/synthetic_data_module.yaml -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/entrypoint_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/entrypoint_test.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/launchers/inference_cond_fip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/launchers/inference_cond_fip.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/launchers/train_cond_fip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/launchers/train_cond_fip.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/launchers/train_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/launchers/train_encoder.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/models/amortized_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/models/amortized_models.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/models/decoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/models/decoder_layer.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/tasks/cond_fip_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/tasks/cond_fip_inference.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/tasks/cond_fip_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/tasks/cond_fip_training.py -------------------------------------------------------------------------------- /research_experiments/cond_fip/src/cond_fip/tasks/encoder_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/cond_fip/src/cond_fip/tasks/encoder_training.py -------------------------------------------------------------------------------- /research_experiments/fip/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/README.md -------------------------------------------------------------------------------- /research_experiments/fip/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/poetry.lock -------------------------------------------------------------------------------- /research_experiments/fip/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/pyproject.toml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/config/amortization/leaf_prediction.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/config/amortization/leaf_prediction.yaml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/config/numpy_tensor_data_module.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/config/numpy_tensor_data_module.yaml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/config/scm_learning_with_ground_truth/scm_learning_true_graph.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/config/scm_learning_with_ground_truth/scm_learning_true_graph.yaml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/config/scm_learning_with_ground_truth/scm_learning_true_perm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/config/scm_learning_with_ground_truth/scm_learning_true_perm.yaml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/config/scm_learning_with_predicted_truth/scm_learning_predicted_leaf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/config/scm_learning_with_predicted_truth/scm_learning_predicted_leaf.yaml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/config/synthetic_data_module.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/config/synthetic_data_module.yaml -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/README.md -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/avici_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/avici_data.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/avici_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/avici_data.sh -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/config_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/config_data.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/csuite_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/csuite_data.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/csuite_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/csuite_data.sh -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/normalizing_flow_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/normalizing_flow_data.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/normalizing_flow_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/normalizing_flow_data.sh -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_generation/sem_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_generation/sem_factory.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_modules/numpy_tensor_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_modules/numpy_tensor_data_module.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/data_modules/synthetic_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/data_modules/synthetic_data_module.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/entrypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/entrypoint.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/launchers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/launchers/README.md -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/launchers/amortization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/launchers/amortization.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/launchers/basic_commands.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/launchers/basic_commands.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/launchers/scm_learning_with_ground_truth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/launchers/scm_learning_with_ground_truth.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/launchers/scm_learning_with_predicted_truth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/launchers/scm_learning_with_predicted_truth.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/methods/base_method_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/methods/base_method_module.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/methods/jacobian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/methods/jacobian.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/methods/scm_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/methods/scm_learning.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/models/amortized_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/models/amortized_models.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/models/causal_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/models/causal_models.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/my_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/my_cli.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/task_utils/leaf_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/task_utils/leaf_functions.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/task_utils/learnable_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/task_utils/learnable_loss.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/tasks/amortization/leaf_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/tasks/amortization/leaf_prediction.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/tasks/scm_learning_with_ground_truth/scm_learning_true_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/tasks/scm_learning_with_ground_truth/scm_learning_true_graph.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/tasks/scm_learning_with_ground_truth/scm_learning_true_perm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/tasks/scm_learning_with_ground_truth/scm_learning_true_perm.py -------------------------------------------------------------------------------- /research_experiments/fip/src/fip/tasks/scm_learning_with_predicted_truth/scm_learning_predicted_leaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/fip/src/fip/tasks/scm_learning_with_predicted_truth/scm_learning_predicted_leaf.py -------------------------------------------------------------------------------- /research_experiments/scotch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/README.md -------------------------------------------------------------------------------- /research_experiments/scotch/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/poetry.lock -------------------------------------------------------------------------------- /research_experiments/scotch/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/pyproject.toml -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/dataset_generation/example_sdes/lorenz_sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/dataset_generation/example_sdes/lorenz_sde.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/dataset_generation/example_sdes/yeast_glycolysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/dataset_generation/example_sdes/yeast_glycolysis.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/dataset_generation/generate_and_save_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/dataset_generation/generate_and_save_data.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/dataset_generation/generate_trajectories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/dataset_generation/generate_trajectories.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/experiments/dream3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/experiments/dream3.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/experiments/lorenz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/experiments/lorenz.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/experiments/netsim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/experiments/netsim.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/experiments/yeast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/experiments/yeast.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/latent_learning/functional_relationship.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/latent_learning/functional_relationship.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/latent_learning/graph_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/latent_learning/graph_distribution.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/latent_learning/scotch_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/latent_learning/scotch_data_module.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/latent_learning/scotch_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/latent_learning/scotch_module.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/latent_learning/scotch_nns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/latent_learning/scotch_nns.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/scotch_utils/graph_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/scotch_utils/graph_metrics.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/scotch_utils/scotch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/scotch_utils/scotch_utils.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/sdes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/sdes/scotch_sdes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/sdes/scotch_sdes.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/sdes/sde_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/sdes/sde_modules.py -------------------------------------------------------------------------------- /research_experiments/scotch/src/scotch/sdes/sdes_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/research_experiments/scotch/src/scotch/sdes/sdes_core.py -------------------------------------------------------------------------------- /src/causica/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/config/lightning/default_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/config/lightning/default_data.yaml -------------------------------------------------------------------------------- /src/causica/config/lightning/default_gaussian.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/config/lightning/default_gaussian.yaml -------------------------------------------------------------------------------- /src/causica/config/lightning/default_spline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/config/lightning/default_spline.yaml -------------------------------------------------------------------------------- /src/causica/data_generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/data_generation/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/data_generation/generate_data.py -------------------------------------------------------------------------------- /src/causica/data_generation/samplers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/data_generation/samplers/functional_relationships_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/data_generation/samplers/functional_relationships_sampler.py -------------------------------------------------------------------------------- /src/causica/data_generation/samplers/noise_dist_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/data_generation/samplers/noise_dist_sampler.py -------------------------------------------------------------------------------- /src/causica/data_generation/samplers/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/data_generation/samplers/sampler.py -------------------------------------------------------------------------------- /src/causica/data_generation/samplers/sem_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/data_generation/samplers/sem_sampler.py -------------------------------------------------------------------------------- /src/causica/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/datasets/causal_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/causal_dataset.py -------------------------------------------------------------------------------- /src/causica/datasets/causica_dataset_format/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/causica_dataset_format/__init__.py -------------------------------------------------------------------------------- /src/causica/datasets/causica_dataset_format/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/causica_dataset_format/load.py -------------------------------------------------------------------------------- /src/causica/datasets/causica_dataset_format/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/causica_dataset_format/save.py -------------------------------------------------------------------------------- /src/causica/datasets/interventional_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/interventional_data.py -------------------------------------------------------------------------------- /src/causica/datasets/loaded_expert_graph_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/loaded_expert_graph_container.py -------------------------------------------------------------------------------- /src/causica/datasets/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/normalization.py -------------------------------------------------------------------------------- /src/causica/datasets/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/samplers.py -------------------------------------------------------------------------------- /src/causica/datasets/synthetic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/synthetic_dataset.py -------------------------------------------------------------------------------- /src/causica/datasets/tensordict_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/tensordict_utils.py -------------------------------------------------------------------------------- /src/causica/datasets/timeseries_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/timeseries_dataset.py -------------------------------------------------------------------------------- /src/causica/datasets/variable_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/datasets/variable_types.py -------------------------------------------------------------------------------- /src/causica/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/__init__.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/__init__.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/adjacency_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/adjacency_distributions.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/constrained_adjacency_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/constrained_adjacency_distributions.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/edges_per_node_erdos_renyi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/edges_per_node_erdos_renyi.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/enco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/enco.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/erdos_renyi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/erdos_renyi.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/fixed_adjacency_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/fixed_adjacency_distribution.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/geometric_random_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/geometric_random_graph.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/gibbs_dag_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/gibbs_dag_prior.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/scale_free.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/scale_free.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/stochastic_block_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/stochastic_block_model.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/temporal_adjacency_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/temporal_adjacency_distributions.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/three_way.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/three_way.py -------------------------------------------------------------------------------- /src/causica/distributions/adjacency/watts_strogatz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/adjacency/watts_strogatz.py -------------------------------------------------------------------------------- /src/causica/distributions/distribution_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/distribution_module.py -------------------------------------------------------------------------------- /src/causica/distributions/gumbel_binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/gumbel_binary.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/__init__.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/bernoulli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/bernoulli.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/categorical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/categorical.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/joint.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/noise.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/spline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/spline/__init__.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/spline/bayesiains_nsf_rqs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/spline/bayesiains_nsf_rqs.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/spline/rational_quadratic_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/spline/rational_quadratic_transform.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/spline/spline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/spline/spline.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/univariate_cauchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/univariate_cauchy.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/univariate_laplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/univariate_laplace.py -------------------------------------------------------------------------------- /src/causica/distributions/noise/univariate_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/noise/univariate_normal.py -------------------------------------------------------------------------------- /src/causica/distributions/signed_uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/signed_uniform.py -------------------------------------------------------------------------------- /src/causica/distributions/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/transforms/__init__.py -------------------------------------------------------------------------------- /src/causica/distributions/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/transforms/base.py -------------------------------------------------------------------------------- /src/causica/distributions/transforms/joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/transforms/joint.py -------------------------------------------------------------------------------- /src/causica/distributions/transforms/tensor_to_tensordict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/distributions/transforms/tensor_to_tensordict.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/__init__.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/deci_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/deci_functional_relationships.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/do_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/do_functional_relationships.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/functional_relationships.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/heteroscedastic_rff_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/heteroscedastic_rff_functional_relationships.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/linear_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/linear_functional_relationships.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/rff_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/rff_functional_relationships.py -------------------------------------------------------------------------------- /src/causica/functional_relationships/temporal_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/functional_relationships/temporal_functional_relationships.py -------------------------------------------------------------------------------- /src/causica/graph/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/graph/dag_constraint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/graph/dag_constraint.py -------------------------------------------------------------------------------- /src/causica/graph/evaluation_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/graph/evaluation_metrics.py -------------------------------------------------------------------------------- /src/causica/lightning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/lightning/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/callbacks.py -------------------------------------------------------------------------------- /src/causica/lightning/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/cli.py -------------------------------------------------------------------------------- /src/causica/lightning/data_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/lightning/data_modules/basic_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/data_modules/basic_data_module.py -------------------------------------------------------------------------------- /src/causica/lightning/data_modules/deci_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/data_modules/deci_data_module.py -------------------------------------------------------------------------------- /src/causica/lightning/data_modules/synthetic_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/data_modules/synthetic_data_module.py -------------------------------------------------------------------------------- /src/causica/lightning/data_modules/variable_spec_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/data_modules/variable_spec_data.py -------------------------------------------------------------------------------- /src/causica/lightning/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/loggers.py -------------------------------------------------------------------------------- /src/causica/lightning/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/main.py -------------------------------------------------------------------------------- /src/causica/lightning/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/lightning/modules/deci_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/modules/deci_module.py -------------------------------------------------------------------------------- /src/causica/lightning/modules/variable_spec_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/lightning/modules/variable_spec_module.py -------------------------------------------------------------------------------- /src/causica/mlflow_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/mlflow_helpers.py -------------------------------------------------------------------------------- /src/causica/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/__init__.py -------------------------------------------------------------------------------- /src/causica/nn/deci_embed_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/deci_embed_nn.py -------------------------------------------------------------------------------- /src/causica/nn/fip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/fip/__init__.py -------------------------------------------------------------------------------- /src/causica/nn/fip/attention_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/fip/attention_layer.py -------------------------------------------------------------------------------- /src/causica/nn/fip/avici_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/fip/avici_encoder.py -------------------------------------------------------------------------------- /src/causica/nn/fip/decoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/fip/decoder_layer.py -------------------------------------------------------------------------------- /src/causica/nn/fip/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/fip/embeddings.py -------------------------------------------------------------------------------- /src/causica/nn/temporal_embed_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/nn/temporal_embed_nn.py -------------------------------------------------------------------------------- /src/causica/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/sem/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/sem/distribution_parameters_sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/sem/distribution_parameters_sem.py -------------------------------------------------------------------------------- /src/causica/sem/sem_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/sem/sem_distribution.py -------------------------------------------------------------------------------- /src/causica/sem/structural_equation_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/sem/structural_equation_model.py -------------------------------------------------------------------------------- /src/causica/sem/temporal_distribution_parameters_sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/sem/temporal_distribution_parameters_sem.py -------------------------------------------------------------------------------- /src/causica/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/causica/training/auglag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/training/auglag.py -------------------------------------------------------------------------------- /src/causica/training/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/training/evaluation.py -------------------------------------------------------------------------------- /src/causica/training/per_variable_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/training/per_variable_metrics.py -------------------------------------------------------------------------------- /src/causica/triangular_transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/src/causica/triangular_transformations.py -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/data_generation/test_generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/data_generation/test_generate_data.py -------------------------------------------------------------------------------- /test/datasets/test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_datasets.py -------------------------------------------------------------------------------- /test/datasets/test_loaded_expert_graph_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_loaded_expert_graph_container.py -------------------------------------------------------------------------------- /test/datasets/test_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_samplers.py -------------------------------------------------------------------------------- /test/datasets/test_standardizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_standardizer.py -------------------------------------------------------------------------------- /test/datasets/test_synthetic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_synthetic_dataset.py -------------------------------------------------------------------------------- /test/datasets/test_tensordict_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_tensordict_utils.py -------------------------------------------------------------------------------- /test/datasets/test_timeseries_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/datasets/test_timeseries_dataset.py -------------------------------------------------------------------------------- /test/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/distributions/adjacency/test_adjacency_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_adjacency_distributions.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_constrained_adjacency_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_constrained_adjacency_distributions.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_edges_per_node_erdos_renyi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_edges_per_node_erdos_renyi.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_enco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_enco.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_erdos_renyi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_erdos_renyi.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_geometric_random_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_geometric_random_graph.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_gibbs_dag_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_gibbs_dag_prior.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_scale_free.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_scale_free.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_stochastic_block_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_stochastic_block_model.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_three_way.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_three_way.py -------------------------------------------------------------------------------- /test/distributions/adjacency/test_watts_strogatz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/adjacency/test_watts_strogatz.py -------------------------------------------------------------------------------- /test/distributions/noise/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/distributions/noise/spline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/distributions/noise/spline/test_splines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/spline/test_splines.py -------------------------------------------------------------------------------- /test/distributions/noise/test_bernoulli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/test_bernoulli.py -------------------------------------------------------------------------------- /test/distributions/noise/test_categorical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/test_categorical.py -------------------------------------------------------------------------------- /test/distributions/noise/test_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/test_joint.py -------------------------------------------------------------------------------- /test/distributions/noise/test_univariate_cauchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/test_univariate_cauchy.py -------------------------------------------------------------------------------- /test/distributions/noise/test_univariate_laplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/test_univariate_laplace.py -------------------------------------------------------------------------------- /test/distributions/noise/test_univariate_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/noise/test_univariate_normal.py -------------------------------------------------------------------------------- /test/distributions/test_sem_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/test_sem_distribution.py -------------------------------------------------------------------------------- /test/distributions/transforms/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/transforms/test_base.py -------------------------------------------------------------------------------- /test/distributions/transforms/test_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/transforms/test_joint.py -------------------------------------------------------------------------------- /test/distributions/transforms/test_transform_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/distributions/transforms/test_transform_modules.py -------------------------------------------------------------------------------- /test/functional_relationships/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/functional_relationships/test_do_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/functional_relationships/test_do_functional_relationships.py -------------------------------------------------------------------------------- /test/functional_relationships/test_functional_relationships.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/functional_relationships/test_functional_relationships.py -------------------------------------------------------------------------------- /test/integration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/integration/decimodule.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/integration/decimodule.pt -------------------------------------------------------------------------------- /test/integration/test_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/integration/test_evaluation.py -------------------------------------------------------------------------------- /test/integration/test_pytorch_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/integration/test_pytorch_lightning.py -------------------------------------------------------------------------------- /test/integration/test_save_load_csuite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/integration/test_save_load_csuite.py -------------------------------------------------------------------------------- /test/integration/test_timeseries_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/integration/test_timeseries_dataset.py -------------------------------------------------------------------------------- /test/lightning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/lightning/test_basic_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/lightning/test_basic_data_module.py -------------------------------------------------------------------------------- /test/lightning/test_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/lightning/test_callbacks.py -------------------------------------------------------------------------------- /test/lightning/test_loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/lightning/test_loggers.py -------------------------------------------------------------------------------- /test/lightning/test_variable_spec_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/lightning/test_variable_spec_data.py -------------------------------------------------------------------------------- /test/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/metrics/test_graph_eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/metrics/test_graph_eval_metrics.py -------------------------------------------------------------------------------- /test/nn/test_deci_embed_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/nn/test_deci_embed_nn.py -------------------------------------------------------------------------------- /test/nn/test_temporal_embed_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/nn/test_temporal_embed_nn.py -------------------------------------------------------------------------------- /test/sem/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/sem/__init__.py -------------------------------------------------------------------------------- /test/sem/test_sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/sem/test_sem.py -------------------------------------------------------------------------------- /test/sem/test_temporal_distribution_parameters_sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/sem/test_temporal_distribution_parameters_sem.py -------------------------------------------------------------------------------- /test/sem/test_treatment_effects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/sem/test_treatment_effects.py -------------------------------------------------------------------------------- /test/test_dag_constraint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/test_dag_constraint.py -------------------------------------------------------------------------------- /test/test_triangular_transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/test_triangular_transformations.py -------------------------------------------------------------------------------- /test/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/training/test_auglag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/causica/HEAD/test/training/test_auglag.py --------------------------------------------------------------------------------