├── .gitignore ├── datasets ├── __init__.py ├── data_gen.py └── simulation.py ├── environment.yml ├── helpers ├── __init__.py ├── analyze_utils.py ├── base.py ├── config_utils.py ├── evaluation.py ├── tf_utils.py └── train_utils.py ├── log_example.txt ├── models ├── __init__.py ├── fed_dag.py ├── linear │ └── notears.py └── nonlinear │ ├── masked_model.py │ └── masked_nn.py ├── readme.md ├── test.py ├── test_run.sh └── trainers ├── __init__.py └── al_trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/.gitignore -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/data_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/datasets/data_gen.py -------------------------------------------------------------------------------- /datasets/simulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/datasets/simulation.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/environment.yml -------------------------------------------------------------------------------- /helpers/__init__.py: -------------------------------------------------------------------------------- 1 | from .base import BaseLearner -------------------------------------------------------------------------------- /helpers/analyze_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/helpers/analyze_utils.py -------------------------------------------------------------------------------- /helpers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/helpers/base.py -------------------------------------------------------------------------------- /helpers/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/helpers/config_utils.py -------------------------------------------------------------------------------- /helpers/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/helpers/evaluation.py -------------------------------------------------------------------------------- /helpers/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/helpers/tf_utils.py -------------------------------------------------------------------------------- /helpers/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/helpers/train_utils.py -------------------------------------------------------------------------------- /log_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/log_example.txt -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/fed_dag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/models/fed_dag.py -------------------------------------------------------------------------------- /models/linear/notears.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/models/linear/notears.py -------------------------------------------------------------------------------- /models/nonlinear/masked_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/models/nonlinear/masked_model.py -------------------------------------------------------------------------------- /models/nonlinear/masked_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/models/nonlinear/masked_nn.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/readme.md -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/test.py -------------------------------------------------------------------------------- /test_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/test_run.sh -------------------------------------------------------------------------------- /trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/trainers/__init__.py -------------------------------------------------------------------------------- /trainers/al_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ErdunGAO/FedDAG/HEAD/trainers/al_trainer.py --------------------------------------------------------------------------------