Complete pipelines and helper tools for data science and machine learning projects.
Project description
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 -c 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.
➡️ Prerequisite: PyTorch required. Follow the official instructions: PyTorch website
pip install "dragon-ml-toolbox[ML]"
Modules:
constants
data_exploration
ETL_cleaning
ETL_engineering
IO_tools
keys
math_utilities
ML_callbacks
ML_chaining_inference
ML_chaining_utilities
ML_configuration
ML_datasetmaster
ML_evaluation_captum
ML_evaluation_multi
ML_evaluation
ML_inference
ML_models
ML_models_advanced
ML_optimization_pareto
ML_optimization
ML_scaler
ML_sequence_datasetmaster
ML_sequence_evaluation
ML_sequence_inference
ML_sequence_models
ML_trainer
ML_utilities
ML_vision_datasetmaster
ML_vision_evaluation
ML_vision_inference
ML_vision_models
ML_vision_transformers
optimization_tools
path_manager
schema
serde
SQL
utilities
Wrappers for some pytorch_tabular models are available:
pip install "dragon-ml-toolbox[ML,py-tab]"
# Extra Modules:
ML_models_pytab
ML_configuration_pytab
🌳 Ensemble Learning [ensemble]
Comprehensive set of tools for typical data science workflows focused on XGBoost and LightGBM.
pip install "dragon-ml-toolbox[ensemble]"
Modules:
constants
IO_tools
data_exploration
ensemble_evaluation
ensemble_inference
ensemble_learning
ETL_cleaning
ETL_engineering
math_utilities
optimization_tools
path_manager
PSO_optimization
schema
serde
SQL
utilities
🔬 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:
constants
IO_tools
math_utilities
MICE_imputation
serde
VIF_factor
path_manager
utilities
📋 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:
constants
IO_tools
GUI_tools
ensemble_inference
path_manager
schema
serde
🤖 GUI for PyTorch Models [gui-torch]
GUI tools compatible with PyTorch models used for inference.
pip install "dragon-ml-toolbox[gui-torch]"
Modules:
constants
IO_tools
GUI_tools
keys
ML_models
ML_models_advanced
ML_sequence_models
ML_vision_models # Requires: torchvision and Pillow
ML_inference
ML_sequence_inference
ML_vision_inference # Requires: torchvision and Pillow
ML_vision_transformers # Requires: torchvision and Pillow
ML_scaler
path_manager
schema
Usage
After installation, import modules like this:
from ml_tools.serde import serialize_object, deserialize_object
from ml_tools.IO_tools import train_logger
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file dragon_ml_toolbox-19.13.0.tar.gz.
File metadata
- Download URL: dragon_ml_toolbox-19.13.0.tar.gz
- Upload date:
- Size: 293.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.9.18 {"installer":{"name":"uv","version":"0.9.18","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}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a02461d28b2548887885c8ad41ffa06907142d78f99023e029ccf2d4c0c43526
|
|
| MD5 |
7037920de1a24a899849daebe13e20be
|
|
| BLAKE2b-256 |
5611841b012549822784395a760adc9208534eb6b6cf1d9734506fb4d7665357
|
File details
Details for the file dragon_ml_toolbox-19.13.0-py3-none-any.whl.
File metadata
- Download URL: dragon_ml_toolbox-19.13.0-py3-none-any.whl
- Upload date:
- Size: 336.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.9.18 {"installer":{"name":"uv","version":"0.9.18","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}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ec323325e95d8c3c7f8877ab47402c4243a3e5fbfc149e3e2dd0774ec9a97456
|
|
| MD5 |
b231e635116d7b82c0cf65a9520d5e63
|
|
| BLAKE2b-256 |
4bbd8617de6447568ea3d2b73d45c158437f4d93e95993abe460d3e4f752000c
|
