dragon-ml-toolbox 4.1.0

A collection of tools for data science and machine learning projects.

dragon-ml-toolbox

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

Features:

• Modular scripts for data exploration, logging, machine learning, and more.
• Designed for seamless integration as a Git submodule or installable Python package.

Installation

Python 3.10+

Via PyPI

Install the latest stable release from PyPI:

pip install dragon-ml-toolbox

Via GitHub (Editable)

Clone the repository and install in editable mode with optional dependencies:

git clone https://github.com/DrAg0n-BoRn/ML_tools.git
cd ML_tools
pip install -e .

Via conda-forge

Install from the conda-forge channel:

conda install -c conda-forge dragon-ml-toolbox

Modular Installation

📦 Core Machine Learning Toolbox [ML]

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

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

To install the standard CPU-only versions of Torch and Torchvision:

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

⚠️ To make use of GPU acceleration (highly recommended), follow the official instructions: PyTorch website

Modules:

custom_logger
data_exploration
datasetmaster
ensemble_learning
ensemble_inference
ETL_engineering
ML_callbacks
ML_evaluation
ML_trainer
ML_inference
path_manager
PSO_optimization
SQL
RNN_forecast
utilities

🔬 MICE Imputation and Variance Inflation Factor [mice]

⚠️ Important: This group has strict version requirements. It is highly recommended to install this group in a separate virtual environment.

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

Modules:

custom_logger
MICE_imputation
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:

custom_logger
handle_excel
path_manager

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

For GUIs that include plotting functionality, you must also install the [plot] extra.

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

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

Modules:

GUI_tools
ensemble_inference
path_manager

🤖 GUI for PyTorch Models [gui-torch]

For GUIs that include plotting functionality, you must also install the [plot] extra.

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

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

Modules:

GUI_tools
ML_inference
path_manager

🎫 Base Tools [base]

General purpose functions and classes.

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

Modules:

ETL_Engineering
custom_logger
SQL
utilities
path_manager

Usage

After installation, import modules like this:

from ml_tools.utilities import serialize_object, deserialize_object
from ml_tools.custom_logger import custom_logger