Skip to main content

No project description provided

Project description

DeepBridge

Documentation Status CI PyPI version PyPI Downloads

DeepBridge is a comprehensive Python library for advanced machine learning model validation, distillation, and performance analysis. It provides powerful tools to manage experiments, validate models, create more efficient model versions, and conduct in-depth performance evaluations.

Installation

You can install DeepBridge using pip:

pip install deepbridge

Or install from source:

git clone https://github.com/DeepBridge-Validation/DeepBridge.git
cd deepbridge
pip install -e .

Key Features

  • Comprehensive Testing Framework

    • Robustness testing with perturbation analysis
    • Uncertainty quantification using conformal prediction
    • Resilience testing under distribution shifts
    • Hyperparameter importance analysis
  • Model Validation

    • Experiment tracking and management
    • Comprehensive model performance analysis
    • Advanced metric tracking
    • Model versioning support
  • Model Distillation

    • Knowledge distillation across multiple model types
    • Automated distillation with hyperparameter optimization
    • Support for GBM, XGBoost, and neural networks
    • Performance optimization and model compression
  • Advanced Analytics & Reporting

    • Interactive HTML reports with Plotly visualizations
    • Static reports for documentation
    • Detailed performance metrics and analysis
    • Multi-model comparison capabilities
  • Synthetic Data Generation

    • Gaussian Copula method
    • Privacy-preserving data synthesis
    • Quality metrics and validation
    • Integration with validation pipeline

Quick Start

Model Validation

from deepbridge.core.experiment import Experiment
from deepbridge.db_data import DBDataset

# Create dataset
dataset = DBDataset(
    data=df,
    target_column='target',
    features=['feature1', 'feature2', 'feature3']
)

# Create experiment
experiment = Experiment(
    name='model_validation',
    dataset=dataset,
    models={'my_model': trained_model}
)

# Run validation tests
robustness_results = experiment.run_test('robustness', config='medium')
uncertainty_results = experiment.run_test('uncertainty', config='medium')

# Generate comprehensive report
experiment.generate_report('robustness', output_dir='./reports')

Model Distillation

from deepbridge.distillation import AutoDistiller
from deepbridge.db_data import DBDataset

# Create dataset with predictions
dataset = DBDataset(
    data=df,
    target_column='target',
    features=features,
    prob_cols=['prob_class_0', 'prob_class_1']
)

# Run automated distillation
distiller = AutoDistiller(
    dataset=dataset,
    output_dir='results',
    test_size=0.2,
    n_trials=10
)
results = distiller.run(use_probabilities=True)

Command-Line Interface

# Run model validation
deepbridge validate --dataset data.csv --model model.pkl --tests all

# Generate reports
deepbridge report --results ./results --output ./reports --format interactive

# Train distilled model
deepbridge distill train gbm predictions.csv features.csv -s ./models

# Generate synthetic data
deepbridge synthetic generate --data original.csv --method gaussian_copula --samples 10000

Requirements

  • Python 3.10-3.12
  • Key Dependencies:
    • numpy >= 2.2.3
    • pandas >= 2.2.3
    • scikit-learn >= 1.6.1
    • xgboost >= 2.1.4
    • scipy >= 1.15.1
    • matplotlib >= 3.10.0
    • seaborn >= 0.13.2
    • plotly >= 6.0.0
    • optuna >= 4.2.1
    • jinja2 >= 3.1.5

Documentation

Full documentation is available at: DeepBridge Documentation

Key Documentation Sections

Quick Links

Contributing

We welcome contributions! Please see our contribution guidelines for details on how to submit pull requests, report issues, and contribute to the project.

  1. Fork the repository
  2. Create your feature branch
  3. Commit your changes
  4. Push to the branch
  5. Open a Pull Request

Recent Updates

  • 2025-07-02: Added comprehensive documentation including Implementation Guide, Testing Framework, Report Generation, and complete API Reference
  • 2025-05-15: Fixed static report chart URLs to properly use relative paths with ./ prefix for improved portability across different environments

Development Setup

# Clone the repository
git clone https://github.com/DeepBridge-Validation/DeepBridge.git
cd deepbridge

# Create virtual environment
python -m venv venv
source venv/bin/activate

# Install dependencies
pip install -r requirements.txt

Running Tests

pytest tests/

License

MIT License

Citation

If you use DeepBridge in your research, please cite:

@software{deepbridge2025,
  title = {DeepBridge: Advanced Model Validation and Distillation Library},
  author = {Gustavo Haase, Paulo Dourado},
  year = {2025},
  url = {https://github.com/DeepBridge-Validation/DeepBridge}
}

Contact

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

deepbridge-0.1.45.tar.gz (986.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

deepbridge-0.1.45-py3-none-any.whl (1.3 MB view details)

Uploaded Python 3

File details

Details for the file deepbridge-0.1.45.tar.gz.

File metadata

  • Download URL: deepbridge-0.1.45.tar.gz
  • Upload date:
  • Size: 986.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.1.3 CPython/3.12.10 Linux/6.6.87.1-microsoft-standard-WSL2

File hashes

Hashes for deepbridge-0.1.45.tar.gz
Algorithm Hash digest
SHA256 9b37e2914f86ba3a82adbfa5bbaf06ac2905e708def4a754c39b64dd351cf112
MD5 6179252a7312949c21159add3f584235
BLAKE2b-256 cfd9822509ca14483c3a253020e64316d0c77e449d9b2346b58cab8b9ea8dac6

See more details on using hashes here.

File details

Details for the file deepbridge-0.1.45-py3-none-any.whl.

File metadata

  • Download URL: deepbridge-0.1.45-py3-none-any.whl
  • Upload date:
  • Size: 1.3 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.1.3 CPython/3.12.10 Linux/6.6.87.1-microsoft-standard-WSL2

File hashes

Hashes for deepbridge-0.1.45-py3-none-any.whl
Algorithm Hash digest
SHA256 77450eb1858e166d6ce90d8d824743b5dbace767e1dbc1b9131037339284d3c4
MD5 dc89b248cd64a4654946e60b5fd8f818
BLAKE2b-256 a4f9ee65de41121312d9581d9f0751c8fdf4f93db5ac2a47faee3b36021e1f09

See more details on using hashes here.

Supported by

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