Skip to main content

dragon-ml-toolbox

A collection of machine learning pipelines and utilities, structured as modular packages for easy reuse and installation. This package has no base dependencies, allowing for lightweight and customized virtual environments.

Features:

  • Modular scripts for data science workflows, including data exploration, ETL, model training, evaluation, and inference.
  • Support for PyTorch-based models, ensemble learning (XGBoost, LightGBM), and MICE imputation.

Installation

Python 3.12

Via PyPI

Install the latest stable release from PyPI:

Using pip:

pip install dragon-ml-toolbox

Using UV:

uv add dragon-ml-toolbox

Via conda-forge

Install from the conda-forge channel:

conda install conda-forge::dragon-ml-toolbox

Modular Installation

This toolbox is designed as a collection of mutually exclusive environments due to conflicting core dependencies.

  • Rule: Create a fresh virtual environment for each module to use.

📦 Core Machine Learning Toolbox [ML]

Installs a comprehensive set of tools for typical data science workflows, including data manipulation, modeling, and evaluation using PyTorch.

➡️ On Windows, the default installation includes the CPU version of PyTorch. Follow the official instructions to install the CUDA version: PyTorch website

pip install "dragon-ml-toolbox[ML]"

Modules:

data_exploration
ETL_cleaning
ETL_engineering
imputation
IO_tools
keys
math_utilities
ML_callbacks
ML_chain
ML_configuration
ML_datasetmaster
ML_evaluation
ML_evaluation_captum
ML_finalize_handler
ML_finetune
ML_inference
ML_inference_sequence
ML_inference_vision
ML_loss
ML_models
ML_models_diffusion
ML_models_sequence
ML_models_vision
ML_optimization
ML_scaler
ML_trainer
ML_trainer_parallel
ML_utilities
ML_vision_transformers
ML_vision_utilities
optimization_tools
outlier_detection
path_manager
plot_fonts
resampling
schema
serde
SQL
utilities
VIF
constants

🌳 Ensemble Learning [ensemble]

Comprehensive set of tools for typical data science workflows focused on XGBoost and LightGBM.

pip install "dragon-ml-toolbox[ensemble]"

Modules:

data_exploration
ensemble_evaluation
ensemble_inference
ensemble_learning
ensemble_optimization
ETL_cleaning
ETL_engineering
imputation
IO_tools
math_utilities
optimization_tools
outlier_detection
path_manager
plot_fonts
resampling
schema
serde
SQL
utilities
VIF
constants

🔬 MICE Imputation and Variance Inflation Factor [mice]

Utilities for advanced data cleaning and statistical checks. Features Multiple Imputation by Chained Equations (MICE) for handling missing data and Variance Inflation Factor (VIF) analysis to detect multicollinearity in features.

pip install "dragon-ml-toolbox[mice]"

Modules:

imputation
IO_tools
math_utilities
MICE
path_manager
plot_fonts
serde
utilities
VIF

📋 Excel File Handling [excel]

Installs dependencies required to process and handle .xlsx or .xls files.

pip install "dragon-ml-toolbox[excel]"

Modules:

IO_tools
excel_handler
path_manager

🎰 GUI for Boosting Algorithms (XGBoost, LightGBM) [gui-boost]

GUI tools compatible with XGBoost and LightGBM models used for inference.

pip install "dragon-ml-toolbox[gui-boost]"

Modules:

ensemble_inference
GUI_tools
IO_tools
path_manager
schema
serde
constants

🤖 GUI for PyTorch Models [gui-torch]

GUI tools compatible with PyTorch models used for inference.

pip install "dragon-ml-toolbox[gui-torch]"

Modules:

GUI_tools
IO_tools
keys
ML_models
ML_models_sequence
ML_models_vision # Requires: torchvision and Pillow
ML_inference
ML_inference_sequence
ML_inference_vision # Requires: torchvision and Pillow
ML_vision_transformers # Requires: torchvision and Pillow
ML_vision_utilities # Requires: torchvision and Pillow
ML_scaler
path_manager
schema
constants

Usage

After installation, import modules like this:

from dragon.serde import serialize_object, deserialize_object
from dragon.IO_tools import custom_logger

Metadata

Release files for dragon-ml-toolbox 27.1.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for dragon-ml-toolbox 27.1.0
File Size Uploaded Provenance
dragon_ml_toolbox-27.1.0.tar.gz 468.6 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for dragon-ml-toolbox 27.1.0
File Interpreter ABI Platform
dragon_ml_toolbox-27.1.0-py3-none-any.whl Python 3 none any Details

Total release size: 1.1 MB

Release files / dragon_ml_toolbox-27.1.0.tar.gz

Download URL dragon_ml_toolbox-27.1.0.tar.gz
Size 468.6 kB
Tags Source
SHA-256 checksum
How to use checksums
ddb07ea498941a6b7634c2647909080fd9565273896c17b9685709c57dcc22f4
BLAKE2b-256 checksum
How to use checksums
1c97f2c6dc86b47a5074738c24c269fa9b11ad3c01e23264339def9253c98455
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

Release files / dragon_ml_toolbox-27.1.0-py3-none-any.whl

Download URL dragon_ml_toolbox-27.1.0-py3-none-any.whl
Size 611.3 kB
Tags Python 3
SHA-256 checksum
How to use checksums
b0f9858bce4add907182869054cf4416db1fb7b5ab61a393aa86ecc6a73327ea
BLAKE2b-256 checksum
How to use checksums
124f649fc08bcb1d58f76b1da4956e1b41c3c2db1cde34f1050a9d060fe04fb6
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

Security for dragon-ml-toolbox 27.1.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

Release history Release notifications | RSS feed

This release

27.1.0 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page