datasage-ml 0.1.0

A simple baseline machine learning toolkit for tabular datasets.

DataSage ML

DataSage ML is a simple open-source Python package for building quick baseline machine learning models on tabular datasets.

It is designed for data science beginners, analysts, and machine learning learners who want a clean starting point for classification projects.

Features

• Load CSV files or pandas DataFrames
• Detect numeric and categorical columns automatically
• Handle missing values
• Encode categorical variables safely
• Train baseline classification models
• Compare model performance
• Select the best model by F1 score
• Generate predictions
• Save the best trained model

Installation for local development

python -m venv .venv
.venv\Scripts\activate
pip install -e .[dev]

For PowerShell, if the extras command gives issues, use:

pip install -e .
pip install pytest build twine

Demo Usage

from datasage_ml import AutoClassifier

clf = AutoClassifier(target="churn")
clf.fit("data/customer_churn.csv")

print(clf.leaderboard())

clf.save_best("best_model.pkl")

Run the example

python examples\demo_classification.py

Run tests

python -m pytest

GitHub push

git init
git add .
git commit -m "Initial commit: add DataSage ML package"
git branch -M main
git remote add origin https://github.com/Jubril-Olasunkanmi/datasage-ml.git
git push -u origin main

Project Structure

datasage-ml/
├── src/
│   └── datasage_ml/
│       ├── __init__.py
│       └── classifier.py
├── examples/
│   └── demo_classification.py
├── data/
│   └── customer_churn.csv
├── tests/
│   └── test_classifier.py
├── pyproject.toml
├── README.md
├── LICENSE
├── CONTRIBUTING.md
├── requirements.txt
└── .gitignore

Author

Ammar Jubril
Technology Consultant | Data Science | Machine Learning | Financial Engineering

GitHub: https://github.com/Jubril-Olasunkanmi