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.40.tar.gz (748.9 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.40-py3-none-any.whl (1.0 MB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: deepbridge-0.1.40.tar.gz
  • Upload date:
  • Size: 748.9 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.40.tar.gz
Algorithm Hash digest
SHA256 85f2413e34269bc349327e4f04c9b2a2ce63cfe4e31544053d19bf86dae08d18
MD5 9ca68c97d5dff724903699b3f41b7221
BLAKE2b-256 fdcfd255b752385ded64207bf4bf512074f58d535cbe4805aa04d6721bf06210

See more details on using hashes here.

File details

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

File metadata

  • Download URL: deepbridge-0.1.40-py3-none-any.whl
  • Upload date:
  • Size: 1.0 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.40-py3-none-any.whl
Algorithm Hash digest
SHA256 3fcac2fbaaa6bfbe80c23ad9404a554be355d09f55379ad686784529ccc9f6e9
MD5 e7a9a0b257f7c568bce9528e55c5a434
BLAKE2b-256 67658fc076dba12b36dc3715698f377f2f2d0631f7afd400a0da0604c4400dc9

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