Skip to main content

data modeling and analysis tool based on Bayesian networks

Project description

BAMT framework logo

BAMT - Bayesian Analytical and Modelling Toolkit

Repository of a data modeling and analysis tool based on Bayesian networks.

Badges

team ITMO NCCR
package pypi Supported Python Versions Supported Python Versions
tests Build coverage
docs Documentation Status
license license
stats downloads downloads/month downloads/week
style Black

Introduction

BAMT - Bayesian Analytical and Modelling Toolkit. This repository contains a data modeling and analysis tool based on Bayesian networks. It can be divided into two main parts - algorithms for constructing and training Bayesian networks on data and algorithms for applying Bayesian networks for filling gaps, generating synthetic data, assessing edge strength, etc.

bamt readme scheme

Installation

BAMT package is available via PyPi:

pip install bamt

BAMT Features

The following algorithms for Bayesian Networks learning are implemented:

  • Building the structure of a Bayesian network based on expert knowledge by directly specifying the structure of the network.
  • Building the structure of a Bayesian network on data using three algorithms - Hill Climbing, evolutionary, and PC (PC is currently under development). For Hill Climbing, the following score functions are implemented - MI, K2, BIC, AIC. The algorithms work on both discrete and mixed data.
  • Learning the parameters of distributions in the nodes of the network based on Gaussian distribution and Mixture Gaussian distribution with automatic selection of the number of components.
  • Non-parametric learning of distributions at nodes using classification and regression models.
  • BigBraveBN - algorithm for structural learning of Bayesian networks with a large number of nodes. Tested on networks with up to 500 nodes.

Difference from existing implementations:

  • Algorithms work on mixed data.
  • Structural learning implements score-functions for mixed data.
  • Parametric learning implements the use of a mixture of Gaussian distributions to approximate continuous distributions.
  • Non-parametric learning of distributions with various user-specified regression and classification models.
  • The algorithm for structural training of large Bayesian networks (> 10 nodes) is based on local training of small networks with their subsequent algorithmic connection.

bn example gif

For example, in terms of data analysis and modeling using Bayesian networks, a pipeline has been implemented to generate synthetic data by sampling from Bayesian networks.

synthetics generation

How to use

Then the necessary classes are imported from the library:

from bamt.networks.hybrid_bn import HybridBN

Next, a network instance is created and training (structure and parameters) is performed:

bn = HybridBN(has_logit=False, use_mixture=True)
bn.add_edges(preprocessed_data)
bn.fit_parameters(data)

Examples & Tutorials

More examples can be found in Documentation.

Publications about BAMT

We have published several articles about BAMT:

Project structure

The latest stable version of the library is available in the master branch.

It includes the following modules and directories:

  • bamt - directory with the framework code:
    • Preprocessing - module for data preprocessing
    • Networks - module for building and training Bayesian networks
    • Nodes - module for nodes support of Bayesian networks
    • Utilities - module for mathematical and graph utilities
  • data - directory with data for experiments and tests
  • tests - directory with unit and integration tests
  • tutorials - directory with tutorials
  • docs - directory with RTD documentation

Preprocessing

Preprocessor module allows users to transform data according to the pipeline (similar to the pipeline in scikit-learn).

Networks

Three types of networks are implemented:

  • HybridBN - Bayesian network with mixed data
  • DiscreteBN - Bayesian network with discrete data
  • ContinuousBN - Bayesian network with continuous data

They are inherited from the abstract class BaseNetwork.

Nodes

Contains classes for nodes of Bayesian networks.

Utilities

Utilities module contains mathematical and graph utilities to support the main functionality of the library.

Web-BAMT

A web interface for BAMT is currently under development. The repository is available at web-BAMT.

Contacts

If you have questions or suggestions, you can contact us at the following address: ideeva@itmo.ru (Irina Deeva)

Our resources:

Citation

@misc{BAMT,
  author={BAMT},
  title = {Repository experiments and data},
  year = {2021},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/ITMO-NSS-team/BAMT.git}},
  url = {https://github.com/ITMO-NSS-team/BAMT.git}
}

@article{deeva2023advanced,
  title={Advanced Approach for Distributions Parameters Learning in Bayesian Networks with Gaussian Mixture Models and Discriminative Models},
  author={Deeva, Irina and Bubnova, Anna and Kalyuzhnaya, Anna V},
  journal={Mathematics},
  volume={11},
  number={2},
  pages={343},
  year={2023},
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

bamt-1.2.72.tar.gz (65.6 kB view details)

Uploaded Source

Built Distribution

bamt-1.2.72-py3-none-any.whl (87.5 kB view details)

Uploaded Python 3

File details

Details for the file bamt-1.2.72.tar.gz.

File metadata

  • Download URL: bamt-1.2.72.tar.gz
  • Upload date:
  • Size: 65.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.3 CPython/3.12.7 Linux/6.11.3-arch1-1

File hashes

Hashes for bamt-1.2.72.tar.gz
Algorithm Hash digest
SHA256 56bed5bd738ad883aeb3bf8e15a930a50c2bb475d1352401cb9f6d2ed0b59523
MD5 45a0c535c53ef04c2ca93b2eed7dc9b3
BLAKE2b-256 1e1a07fcc68b4bafef8c9787961b7386806bccd786864559fc09d0ade8168cb5

See more details on using hashes here.

File details

Details for the file bamt-1.2.72-py3-none-any.whl.

File metadata

  • Download URL: bamt-1.2.72-py3-none-any.whl
  • Upload date:
  • Size: 87.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.3 CPython/3.12.7 Linux/6.11.3-arch1-1

File hashes

Hashes for bamt-1.2.72-py3-none-any.whl
Algorithm Hash digest
SHA256 9edd7048610f3ddc956cfb7f9e941842f7339dcf40dfcd89b33394d98d8f54f3
MD5 cce1381c61a981c21690fc8b070e57cf
BLAKE2b-256 fe36ade1c7710b55ce035b5dc884da2ed1866be6e21833d16b93d0ec40830cb4

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page