A production-ready CLI toolkit for training, evaluating, and tracking Machine Learning and Deep Learning models with experiment tracking, hyperparameter tuning, model explainability, and an interactive TUI
Project description
███╗ ███╗██╗ ██████╗██╗ ██╗
████╗ ████║██║ ██╔════╝██║ ██║
██╔████╔██║██║ ██║ ██║ ██║
██║╚██╔╝██║██║ ██║ ██║ ██║
██║ ╚═╝ ██║███████╗╚██████╗███████╗██║
╚═╝ ╚═╝╚══════╝ ╚═════╝╚══════╝╚═╝
MLCLI - Machine Learning Command Line Interface
A powerful, modular CLI tool for training, evaluating, and tracking ML/DL models
✨ Features
- ML Models: Logistic Regression, SVM, Random Forest, XGBoost
- DL Models: TensorFlow DNN, CNN, RNN/LSTM/GRU
- Hyperparameter Tuning: Grid Search, Random Search, Bayesian (Optuna)
- Model Explainability: SHAP & LIME
- Preprocessing: Scalers, Normalizers, Encoders, Feature Selection
- Experiment Tracking: Built-in tracker with JSON storage
- Interactive TUI: Terminal-based user interface
🚀 Quick Start
Install
pip install mlcli-toolkit
Verify
mlcli --help
Train a Model
mlcli train --config configs/rf_config.json
Launch Interactive UI
mlcli ui
📋 Commands
| Command | Description |
|---|---|
mlcli list-models |
List available model trainers |
mlcli train -c <config> |
Train a model |
mlcli eval -m <model> -d <data> -t <type> |
Evaluate a model |
mlcli tune -c <config> -m <method> |
Hyperparameter tuning |
mlcli explain -m <model> -d <data> -e <method> |
Model explainability |
mlcli preprocess -d <data> -o <output> -m <method> |
Preprocess data |
mlcli list-runs |
List experiment runs |
mlcli ui |
Launch interactive TUI |
📝 Configuration Example
{
"model": {
"type": "random_forest",
"params": {
"n_estimators": 100,
"max_depth": null,
"random_state": 42
}
},
"dataset": {
"path": "data/train.csv",
"type": "csv",
"target_column": "target"
},
"training": {
"test_size": 0.2,
"random_state": 42
},
"output": {
"model_dir": "artifacts",
"save_formats": ["pickle", "joblib"]
}
}
📚 Documentation
For complete documentation including:
- All configuration options
- Hyperparameter tuning guides
- Model explainability (SHAP/LIME)
- Data preprocessing pipeline
- Extending MLCLI with custom trainers
- Troubleshooting
🤝 Contributing
Contributions are welcome! Please feel free to submit a Pull Request.
📄 License
This project is licensed under the MIT License.
Project details
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 mlcli_toolkit-0.3.1.tar.gz.
File metadata
- Download URL: mlcli_toolkit-0.3.1.tar.gz
- Upload date:
- Size: 96.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
e52c9646bb0f7c34ced5c4ef938a9ea8fa3c4b9051bd36816c69452db4dc0ea7
|
|
| MD5 |
e3cd31ea3dc15086507f38008b561763
|
|
| BLAKE2b-256 |
1a8baef56ffea3673b7fda7c9c025670923555ab77d7f7d924b593054400e212
|
File details
Details for the file mlcli_toolkit-0.3.1-py3-none-any.whl.
File metadata
- Download URL: mlcli_toolkit-0.3.1-py3-none-any.whl
- Upload date:
- Size: 140.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
df15294df7e9d11244de316b406cc18a4622be047ef575b34b113cbb5d9a7966
|
|
| MD5 |
73e031742a2af4288931ba18e9aa8b23
|
|
| BLAKE2b-256 |
b88f73457a11d640f960d3bbb116da0f7b36751843f3d1b443b85547c6dab08c
|