deepbridge 0.1.59

DeepBridge: Framework for ML Model Validation and Knowledge Distillation

DeepBridge

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
  • Fairness testing and bias detection (NEW!)
    • 15 fairness metrics (pre-training and post-training)
    • Auto-detection of sensitive attributes
    • EEOC compliance verification (80% rule)
    • Threshold analysis for fairness optimization
    • Interactive HTML reports with visualizations

• 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)

Fairness Testing

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

# Create dataset (model already trained)
dataset = DBDataset(
    data=df,
    target_column='approved',
    model=trained_model
)

# Create experiment with protected attributes
experiment = Experiment(
    dataset=dataset,
    experiment_type="binary_classification",
    tests=["fairness"],
    protected_attributes=['gender', 'race', 'age_group']
)

# Run fairness tests
fairness_result = experiment.run_fairness_tests(config='full')

# Check results
print(f"Overall Fairness Score: {fairness_result.overall_fairness_score:.3f}")
print(f"Critical Issues: {len(fairness_result.critical_issues)}")
print(f"EEOC Compliant: {fairness_result.overall_fairness_score >= 0.80}")

# Generate interactive HTML report
fairness_result.save_html('fairness_report.html', model_name='My Model')

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

• Getting Started - Installation and basic examples
• User Guide - Core concepts and tutorials
• Technical Reference - Architecture and implementation details
• API Reference - Complete API documentation

Quick Links

• Installation Guide
• Basic Examples
• Complete Workflow
• FAQ
• Troubleshooting

Fairness Documentation

• Fairness Tutorial (Step-by-Step) - Complete tutorial from basics to production
• Best Practices Guide - Guidelines for ethical ML and fairness
• FAQ - Common questions and troubleshooting
• Complete Example - End-to-end executable example

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-11-03: NEW Fairness Module - Complete fairness testing framework with 15 metrics, auto-detection of sensitive attributes, EEOC compliance checks, threshold analysis, and interactive HTML reports. Includes comprehensive documentation, tutorial, and examples.
• 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

• GitHub Issues: DeepBridge Issues
• Email: [e-mail]