├── .gitignore ├── LICENSE ├── README.md ├── docs ├── README.md ├── appendix │ ├── tips.md │ └── tips_md │ │ └── example.jpg ├── build_docs.sh ├── ds │ ├── brn.md │ ├── overview.md │ ├── rn.md │ └── usecases.md ├── formatting │ └── extra.css ├── index.md ├── install │ └── install.md ├── reference │ ├── models.md │ ├── modules.md │ ├── nn.md │ └── utils.md ├── static │ ├── torchlogic_logo.png │ └── torchlogic_logo_black.png └── tutorials │ └── brn.md ├── experiments ├── NRN_Paper_Plots_AAAI.ipynb ├── README.md ├── explain.py ├── explain_global.py ├── explain_nam.py ├── incremental_deletion_RF.py ├── incremental_deletion_RF_SHAP.py ├── incremental_deletion_RNRN.py ├── incremental_deletion_nam.py ├── main.py ├── nam_shap_size.py ├── poetry.lock ├── pyproject.toml ├── query_openml.py └── src │ ├── BRCG.py │ ├── LEURN │ ├── Causality_Example.png │ ├── DATA.py │ ├── DEMO.py │ ├── LEURN.py │ ├── LICENSE │ ├── Presentation_Product.pdf │ ├── Presentation_Technical.pdf │ ├── README.md │ ├── TRAINER.py │ ├── UI.py │ ├── credit_scoring.csv │ └── requirements.txt │ ├── __init__.py │ ├── cofrnet.py │ ├── danet │ ├── DAN_Task.py │ ├── Figures │ │ └── DAN.jpg │ ├── LICENSE │ ├── README.md │ ├── __init__.py │ ├── abstract_model.py │ ├── config │ │ ├── MSLR.yaml │ │ ├── cardio.yaml │ │ ├── click.yaml │ │ ├── default.py │ │ ├── epsilon.yaml │ │ ├── forest_cover_type.yaml │ │ ├── yahoo.yaml │ │ └── year.yaml │ ├── data │ │ ├── data_util.py │ │ └── dataset.py │ ├── main.py │ ├── model │ │ ├── AcceleratedModule.py │ │ ├── DANet.py │ │ ├── __init__.py │ │ └── sparsemax │ │ │ ├── __init__.py │ │ │ └── sparsemax.py │ ├── predict.py │ └── requirements.txt │ ├── danet_model.py │ ├── datasets.py │ ├── difflogic │ ├── INSTALLATION_SUPPORT.md │ ├── LICENSE │ ├── README.md │ ├── difflogic │ │ ├── __init__.py │ │ ├── compiled_model.py │ │ ├── cuda │ │ │ ├── difflogic.cpp │ │ │ └── difflogic_kernel.cu │ │ ├── difflogic.py │ │ ├── functional.py │ │ └── packbitstensor.py │ ├── difflogic_logo.png │ ├── experiments │ │ ├── apply_compiled_net.py │ │ ├── main.py │ │ ├── main_baseline.py │ │ ├── mnist_dataset.py │ │ ├── requirements.txt │ │ ├── results_json.py │ │ └── uci_datasets.py │ └── setup.py │ ├── difflogic_model.py │ ├── encoders.py │ ├── fttransformer.py │ ├── iris_datasets.py │ ├── mlp.py │ ├── nam.py │ ├── neural_additive_models │ ├── README.md │ ├── __init__.py │ ├── data_utils.py │ ├── graph_builder.py │ ├── models.py │ ├── nam_train.py │ ├── nam_train_test.py │ ├── requirements.txt │ ├── run.sh │ ├── setup.py │ └── tests │ │ ├── __init__.py │ │ ├── data_utils_test.py │ │ ├── graph_builder_test.py │ │ └── models_test.py │ └── tuners.py ├── mkdocs.yml ├── notebooks └── tutorials │ ├── Bandit-NRN Tutorial - Binary-Class.ipynb │ ├── Bandit-NRN Tutorial - Domain Knowledge.ipynb │ ├── Bandit-NRN Tutorial - Multi-Class.ipynb │ ├── Bandit-NRN Tutorial - Regression.ipynb │ └── configs │ └── logging.yaml ├── poetry.lock ├── project.yaml ├── pyproject.toml ├── setup.py ├── tests ├── README.md ├── models │ ├── test__boosted_brrn.py │ ├── test__brrn.py │ ├── test_mixin_classifier.py │ └── test_mixin_regressor.py ├── modules │ ├── test_attn.py │ ├── test_brnn.py │ └── test_var.py ├── nn │ ├── test_base_blocks.py │ ├── test_base_predicates.py │ ├── test_base_utils.py │ ├── test_forward_blocks.py │ └── test_forward_utils.py ├── sklogic │ ├── test__base__estimator.py │ ├── test__rnrn_classifier.py │ └── test__rnrn_regressor.py └── utils │ ├── test_base_trainer.py │ ├── test_boosted_brrn_trainer.py │ ├── test_brrn_trainer.py │ ├── test_explanations.py │ └── test_simplification.py └── torchlogic ├── __init__.py ├── models ├── __init__.py ├── attn_classifier.py ├── attn_regressor.py ├── base │ ├── __init__.py │ ├── attn.py │ ├── boosted_brn.py │ ├── brn.py │ ├── pruningrn.py │ ├── rn.py │ └── var.py ├── brn_classifier.py ├── brn_regressor.py ├── mixins │ ├── __init__.py │ ├── _base_mixin.py │ ├── classifier.py │ └── regressor.py ├── var_classifier.py └── var_regressor.py ├── modules ├── __init__.py ├── attn.py ├── brn.py └── var.py ├── nn ├── __init__.py ├── base │ ├── __init__.py │ ├── _core.py │ ├── blocks.py │ ├── constants.py │ ├── predicates.py │ └── utils.py ├── blocks.py ├── predicates.py └── utils.py ├── sklogic ├── __init__.py ├── base │ ├── __init__.py │ └── base_estimator.py ├── classifiers │ ├── RNRNClassifier.py │ └── __init__.py ├── datasets │ ├── __init__.py │ └── simple_dataset.py └── regressors │ ├── RNRNRegressor.py │ └── __init__.py └── utils ├── __init__.py ├── distributed.py ├── explanations ├── __init__.py ├── explanations.py └── simplification.py ├── operations.py └── trainers ├── __init__.py ├── attnnrntrainer.py ├── banditnrntrainer.py ├── base ├── __init__.py └── basetrainer.py └── boostedbanditnrntrainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/README.md -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/appendix/tips.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/appendix/tips.md -------------------------------------------------------------------------------- /docs/appendix/tips_md/example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/appendix/tips_md/example.jpg -------------------------------------------------------------------------------- /docs/build_docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/build_docs.sh -------------------------------------------------------------------------------- /docs/ds/brn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/ds/brn.md -------------------------------------------------------------------------------- /docs/ds/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/ds/overview.md -------------------------------------------------------------------------------- /docs/ds/rn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/ds/rn.md -------------------------------------------------------------------------------- /docs/ds/usecases.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/ds/usecases.md -------------------------------------------------------------------------------- /docs/formatting/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/formatting/extra.css -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/install/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/install/install.md -------------------------------------------------------------------------------- /docs/reference/models.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/reference/models.md -------------------------------------------------------------------------------- /docs/reference/modules.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/reference/modules.md -------------------------------------------------------------------------------- /docs/reference/nn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/reference/nn.md -------------------------------------------------------------------------------- /docs/reference/utils.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/reference/utils.md -------------------------------------------------------------------------------- /docs/static/torchlogic_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/static/torchlogic_logo.png -------------------------------------------------------------------------------- /docs/static/torchlogic_logo_black.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/static/torchlogic_logo_black.png -------------------------------------------------------------------------------- /docs/tutorials/brn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/docs/tutorials/brn.md -------------------------------------------------------------------------------- /experiments/NRN_Paper_Plots_AAAI.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/NRN_Paper_Plots_AAAI.ipynb -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- 1 | # cao-exps -------------------------------------------------------------------------------- /experiments/explain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/explain.py -------------------------------------------------------------------------------- /experiments/explain_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/explain_global.py -------------------------------------------------------------------------------- /experiments/explain_nam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/explain_nam.py -------------------------------------------------------------------------------- /experiments/incremental_deletion_RF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/incremental_deletion_RF.py -------------------------------------------------------------------------------- /experiments/incremental_deletion_RF_SHAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/incremental_deletion_RF_SHAP.py -------------------------------------------------------------------------------- /experiments/incremental_deletion_RNRN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/incremental_deletion_RNRN.py -------------------------------------------------------------------------------- /experiments/incremental_deletion_nam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/incremental_deletion_nam.py -------------------------------------------------------------------------------- /experiments/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/main.py -------------------------------------------------------------------------------- /experiments/nam_shap_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/nam_shap_size.py -------------------------------------------------------------------------------- /experiments/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/poetry.lock -------------------------------------------------------------------------------- /experiments/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/pyproject.toml -------------------------------------------------------------------------------- /experiments/query_openml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/query_openml.py -------------------------------------------------------------------------------- /experiments/src/BRCG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/BRCG.py -------------------------------------------------------------------------------- /experiments/src/LEURN/Causality_Example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/Causality_Example.png -------------------------------------------------------------------------------- /experiments/src/LEURN/DATA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/DATA.py -------------------------------------------------------------------------------- /experiments/src/LEURN/DEMO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/DEMO.py -------------------------------------------------------------------------------- /experiments/src/LEURN/LEURN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/LEURN.py -------------------------------------------------------------------------------- /experiments/src/LEURN/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/LICENSE -------------------------------------------------------------------------------- /experiments/src/LEURN/Presentation_Product.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/Presentation_Product.pdf -------------------------------------------------------------------------------- /experiments/src/LEURN/Presentation_Technical.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/Presentation_Technical.pdf -------------------------------------------------------------------------------- /experiments/src/LEURN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/README.md -------------------------------------------------------------------------------- /experiments/src/LEURN/TRAINER.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/TRAINER.py -------------------------------------------------------------------------------- /experiments/src/LEURN/UI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/UI.py -------------------------------------------------------------------------------- /experiments/src/LEURN/credit_scoring.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/credit_scoring.csv -------------------------------------------------------------------------------- /experiments/src/LEURN/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/LEURN/requirements.txt -------------------------------------------------------------------------------- /experiments/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/src/cofrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/cofrnet.py -------------------------------------------------------------------------------- /experiments/src/danet/DAN_Task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/DAN_Task.py -------------------------------------------------------------------------------- /experiments/src/danet/Figures/DAN.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/Figures/DAN.jpg -------------------------------------------------------------------------------- /experiments/src/danet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/LICENSE -------------------------------------------------------------------------------- /experiments/src/danet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/README.md -------------------------------------------------------------------------------- /experiments/src/danet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/src/danet/abstract_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/abstract_model.py -------------------------------------------------------------------------------- /experiments/src/danet/config/MSLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/MSLR.yaml -------------------------------------------------------------------------------- /experiments/src/danet/config/cardio.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/cardio.yaml -------------------------------------------------------------------------------- /experiments/src/danet/config/click.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/click.yaml -------------------------------------------------------------------------------- /experiments/src/danet/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/default.py -------------------------------------------------------------------------------- /experiments/src/danet/config/epsilon.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/epsilon.yaml -------------------------------------------------------------------------------- /experiments/src/danet/config/forest_cover_type.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/forest_cover_type.yaml -------------------------------------------------------------------------------- /experiments/src/danet/config/yahoo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/yahoo.yaml -------------------------------------------------------------------------------- /experiments/src/danet/config/year.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/config/year.yaml -------------------------------------------------------------------------------- /experiments/src/danet/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/data/data_util.py -------------------------------------------------------------------------------- /experiments/src/danet/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/data/dataset.py -------------------------------------------------------------------------------- /experiments/src/danet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/main.py -------------------------------------------------------------------------------- /experiments/src/danet/model/AcceleratedModule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/model/AcceleratedModule.py -------------------------------------------------------------------------------- /experiments/src/danet/model/DANet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/model/DANet.py -------------------------------------------------------------------------------- /experiments/src/danet/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/src/danet/model/sparsemax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/src/danet/model/sparsemax/sparsemax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/model/sparsemax/sparsemax.py -------------------------------------------------------------------------------- /experiments/src/danet/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet/predict.py -------------------------------------------------------------------------------- /experiments/src/danet/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.4.0 2 | category_encoders 3 | yacs 4 | tensorboard>=2.2.2 5 | qhoptim -------------------------------------------------------------------------------- /experiments/src/danet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/danet_model.py -------------------------------------------------------------------------------- /experiments/src/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/datasets.py -------------------------------------------------------------------------------- /experiments/src/difflogic/INSTALLATION_SUPPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/INSTALLATION_SUPPORT.md -------------------------------------------------------------------------------- /experiments/src/difflogic/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/LICENSE -------------------------------------------------------------------------------- /experiments/src/difflogic/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/README.md -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/__init__.py -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/compiled_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/compiled_model.py -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/cuda/difflogic.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/cuda/difflogic.cpp -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/cuda/difflogic_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/cuda/difflogic_kernel.cu -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/difflogic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/difflogic.py -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/functional.py -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic/packbitstensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic/packbitstensor.py -------------------------------------------------------------------------------- /experiments/src/difflogic/difflogic_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/difflogic_logo.png -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/apply_compiled_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/apply_compiled_net.py -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/main.py -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/main_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/main_baseline.py -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/mnist_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/mnist_dataset.py -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/requirements.txt -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/results_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/results_json.py -------------------------------------------------------------------------------- /experiments/src/difflogic/experiments/uci_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/experiments/uci_datasets.py -------------------------------------------------------------------------------- /experiments/src/difflogic/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic/setup.py -------------------------------------------------------------------------------- /experiments/src/difflogic_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/difflogic_model.py -------------------------------------------------------------------------------- /experiments/src/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/encoders.py -------------------------------------------------------------------------------- /experiments/src/fttransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/fttransformer.py -------------------------------------------------------------------------------- /experiments/src/iris_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/iris_datasets.py -------------------------------------------------------------------------------- /experiments/src/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/mlp.py -------------------------------------------------------------------------------- /experiments/src/nam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/nam.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/README.md -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/__init__.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/data_utils.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/graph_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/graph_builder.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/models.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/nam_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/nam_train.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/nam_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/nam_train_test.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/requirements.txt -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/run.sh -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/setup.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/tests/__init__.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/tests/data_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/tests/data_utils_test.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/tests/graph_builder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/tests/graph_builder_test.py -------------------------------------------------------------------------------- /experiments/src/neural_additive_models/tests/models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/neural_additive_models/tests/models_test.py -------------------------------------------------------------------------------- /experiments/src/tuners.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/experiments/src/tuners.py -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /notebooks/tutorials/Bandit-NRN Tutorial - Binary-Class.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/notebooks/tutorials/Bandit-NRN Tutorial - Binary-Class.ipynb -------------------------------------------------------------------------------- /notebooks/tutorials/Bandit-NRN Tutorial - Domain Knowledge.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/notebooks/tutorials/Bandit-NRN Tutorial - Domain Knowledge.ipynb -------------------------------------------------------------------------------- /notebooks/tutorials/Bandit-NRN Tutorial - Multi-Class.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/notebooks/tutorials/Bandit-NRN Tutorial - Multi-Class.ipynb -------------------------------------------------------------------------------- /notebooks/tutorials/Bandit-NRN Tutorial - Regression.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/notebooks/tutorials/Bandit-NRN Tutorial - Regression.ipynb -------------------------------------------------------------------------------- /notebooks/tutorials/configs/logging.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/notebooks/tutorials/configs/logging.yaml -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/poetry.lock -------------------------------------------------------------------------------- /project.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/project.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/setup.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/models/test__boosted_brrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/models/test__boosted_brrn.py -------------------------------------------------------------------------------- /tests/models/test__brrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/models/test__brrn.py -------------------------------------------------------------------------------- /tests/models/test_mixin_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/models/test_mixin_classifier.py -------------------------------------------------------------------------------- /tests/models/test_mixin_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/models/test_mixin_regressor.py -------------------------------------------------------------------------------- /tests/modules/test_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/modules/test_attn.py -------------------------------------------------------------------------------- /tests/modules/test_brnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/modules/test_brnn.py -------------------------------------------------------------------------------- /tests/modules/test_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/modules/test_var.py -------------------------------------------------------------------------------- /tests/nn/test_base_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/nn/test_base_blocks.py -------------------------------------------------------------------------------- /tests/nn/test_base_predicates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/nn/test_base_predicates.py -------------------------------------------------------------------------------- /tests/nn/test_base_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/nn/test_base_utils.py -------------------------------------------------------------------------------- /tests/nn/test_forward_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/nn/test_forward_blocks.py -------------------------------------------------------------------------------- /tests/nn/test_forward_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/nn/test_forward_utils.py -------------------------------------------------------------------------------- /tests/sklogic/test__base__estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/sklogic/test__base__estimator.py -------------------------------------------------------------------------------- /tests/sklogic/test__rnrn_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/sklogic/test__rnrn_classifier.py -------------------------------------------------------------------------------- /tests/sklogic/test__rnrn_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/sklogic/test__rnrn_regressor.py -------------------------------------------------------------------------------- /tests/utils/test_base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/utils/test_base_trainer.py -------------------------------------------------------------------------------- /tests/utils/test_boosted_brrn_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/utils/test_boosted_brrn_trainer.py -------------------------------------------------------------------------------- /tests/utils/test_brrn_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/utils/test_brrn_trainer.py -------------------------------------------------------------------------------- /tests/utils/test_explanations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/utils/test_explanations.py -------------------------------------------------------------------------------- /tests/utils/test_simplification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/tests/utils/test_simplification.py -------------------------------------------------------------------------------- /torchlogic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchlogic/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/__init__.py -------------------------------------------------------------------------------- /torchlogic/models/attn_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/attn_classifier.py -------------------------------------------------------------------------------- /torchlogic/models/attn_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/attn_regressor.py -------------------------------------------------------------------------------- /torchlogic/models/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/__init__.py -------------------------------------------------------------------------------- /torchlogic/models/base/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/attn.py -------------------------------------------------------------------------------- /torchlogic/models/base/boosted_brn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/boosted_brn.py -------------------------------------------------------------------------------- /torchlogic/models/base/brn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/brn.py -------------------------------------------------------------------------------- /torchlogic/models/base/pruningrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/pruningrn.py -------------------------------------------------------------------------------- /torchlogic/models/base/rn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/rn.py -------------------------------------------------------------------------------- /torchlogic/models/base/var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/base/var.py -------------------------------------------------------------------------------- /torchlogic/models/brn_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/brn_classifier.py -------------------------------------------------------------------------------- /torchlogic/models/brn_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/brn_regressor.py -------------------------------------------------------------------------------- /torchlogic/models/mixins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/mixins/__init__.py -------------------------------------------------------------------------------- /torchlogic/models/mixins/_base_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/mixins/_base_mixin.py -------------------------------------------------------------------------------- /torchlogic/models/mixins/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/mixins/classifier.py -------------------------------------------------------------------------------- /torchlogic/models/mixins/regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/mixins/regressor.py -------------------------------------------------------------------------------- /torchlogic/models/var_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/var_classifier.py -------------------------------------------------------------------------------- /torchlogic/models/var_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/models/var_regressor.py -------------------------------------------------------------------------------- /torchlogic/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/modules/__init__.py -------------------------------------------------------------------------------- /torchlogic/modules/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/modules/attn.py -------------------------------------------------------------------------------- /torchlogic/modules/brn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/modules/brn.py -------------------------------------------------------------------------------- /torchlogic/modules/var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/modules/var.py -------------------------------------------------------------------------------- /torchlogic/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/__init__.py -------------------------------------------------------------------------------- /torchlogic/nn/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/base/__init__.py -------------------------------------------------------------------------------- /torchlogic/nn/base/_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/base/_core.py -------------------------------------------------------------------------------- /torchlogic/nn/base/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/base/blocks.py -------------------------------------------------------------------------------- /torchlogic/nn/base/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/base/constants.py -------------------------------------------------------------------------------- /torchlogic/nn/base/predicates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/base/predicates.py -------------------------------------------------------------------------------- /torchlogic/nn/base/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/base/utils.py -------------------------------------------------------------------------------- /torchlogic/nn/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/blocks.py -------------------------------------------------------------------------------- /torchlogic/nn/predicates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/predicates.py -------------------------------------------------------------------------------- /torchlogic/nn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/nn/utils.py -------------------------------------------------------------------------------- /torchlogic/sklogic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchlogic/sklogic/base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchlogic/sklogic/base/base_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/sklogic/base/base_estimator.py -------------------------------------------------------------------------------- /torchlogic/sklogic/classifiers/RNRNClassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/sklogic/classifiers/RNRNClassifier.py -------------------------------------------------------------------------------- /torchlogic/sklogic/classifiers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/sklogic/classifiers/__init__.py -------------------------------------------------------------------------------- /torchlogic/sklogic/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchlogic/sklogic/datasets/simple_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/sklogic/datasets/simple_dataset.py -------------------------------------------------------------------------------- /torchlogic/sklogic/regressors/RNRNRegressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/sklogic/regressors/RNRNRegressor.py -------------------------------------------------------------------------------- /torchlogic/sklogic/regressors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/sklogic/regressors/__init__.py -------------------------------------------------------------------------------- /torchlogic/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/__init__.py -------------------------------------------------------------------------------- /torchlogic/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/distributed.py -------------------------------------------------------------------------------- /torchlogic/utils/explanations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/explanations/__init__.py -------------------------------------------------------------------------------- /torchlogic/utils/explanations/explanations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/explanations/explanations.py -------------------------------------------------------------------------------- /torchlogic/utils/explanations/simplification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/explanations/simplification.py -------------------------------------------------------------------------------- /torchlogic/utils/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/operations.py -------------------------------------------------------------------------------- /torchlogic/utils/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/trainers/__init__.py -------------------------------------------------------------------------------- /torchlogic/utils/trainers/attnnrntrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/trainers/attnnrntrainer.py -------------------------------------------------------------------------------- /torchlogic/utils/trainers/banditnrntrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/trainers/banditnrntrainer.py -------------------------------------------------------------------------------- /torchlogic/utils/trainers/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/trainers/base/__init__.py -------------------------------------------------------------------------------- /torchlogic/utils/trainers/base/basetrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/trainers/base/basetrainer.py -------------------------------------------------------------------------------- /torchlogic/utils/trainers/boostedbanditnrntrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IBM/torchlogic/HEAD/torchlogic/utils/trainers/boostedbanditnrntrainer.py --------------------------------------------------------------------------------