Skip to main content

Vinzy AutoML - Automated Machine Learning Package by Vinayak Bhosale

Vinzy AutoML is a comprehensive automated machine learning package that simplifies the process of model selection, hyperparameter tuning, and performance evaluation. Designed for both beginners and experienced data scientists, it supports over 60 machine learning models with parallel processing capabilities.


🔑 Key Features

  • 🚀 Automated model selection for regression and classification problems
  • ⚡ Parallel training using multithreading for faster execution
  • 📊 Comprehensive performance metrics with built-in visualization tools
  • 🔍 Hyperparameter tuning with GridSearchCV
  • 🤖 Support for 60+ models including:
    • Scikit-learn models
    • XGBoost, LightGBM, and CatBoost (optional)
  • 🎯 Automatic best model selection based on problem type
  • 🎨 Beautiful visualizations including performance charts and radar plots

📦 Installation

Install the base package:

pip install vinzy_automl

For full functionality including XGBoost, LightGBM, and CatBoost:

pip install vinzy_automl[full]

🧪 Basic Usage

🔹 Regression Example

from vinzy_automl import AutoML
from sklearn.datasets import make_regression
from sklearn.model_selection import train_test_split

# Create sample dataset
X, y = make_regression(n_samples=1000, n_features=20)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# Initialize and train AutoML
automl = AutoML()
results = automl.fit(
    X_train, y_train,
    X_test, y_test,
    problem_type='regression', #or 'classification'
    grid_search=['RandomForest', 'XGBoost'],  # Models to tune
    exclude_models=['Lasso', 'Ridge'],        # Models to skip
    max_workers=4                             # Threads to use
)

# View results
print(results)
# View results in a dataframe
print(automl.get_results_in_dataframe(sort_by='R2'))
automl.plot_performance()

🔹 Classification Example

from vinzy_automl import AutoML
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split

# Create sample dataset
X, y = make_classification(n_samples=1000, n_classes=3)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# Initialize and train AutoML
automl = AutoML()
results = automl.fit(
    X_train, y_train,
    X_test, y_test,
    problem_type='classification', #or 'regression'
    grid_search=['RandomForest', 'XGBoost'],  # Models to tune
    exclude_models=['GaussianNB', 'MultinomialNB'],   # Models to skip
    max_workers=4                             # Threads to use
)

# Get best model and make predictions
best_name, best_model, best_score = automl.get_best_model()
predictions = automl.predict(X_test)

# Visualize results
automl.plot_model_comparison_radar()

To obtain models names use

# To obtain models names use
print(automl.nomenclature())

⚙️ Advanced Features

🔧 Custom Models and Parameters

from sklearn.ensemble import RandomForestRegressor
from sklearn.svm import SVR

# Add custom models
custom_models = {
    'MyRF': RandomForestRegressor(n_estimators=200),
    'MySVR': SVR(kernel='poly')
}

# Custom hyperparameter grids
custom_params = {
    'MyRF': {'max_depth': [5, 10, None]},
    'MySVR': {'C': [0.1, 1, 10], 'degree': [2, 3]}
}

automl.fit(
    X_train, y_train,
    X_test, y_test,
    problem_type='regression',
    custom_models=custom_models,
    custom_param_grids=custom_params
)

🧵 Full Pipeline Training

# Train best model on full dataset
final_model = automl.train_best_model(X_full, y_full)

# Save the trained model
import joblib
joblib.dump(final_model, 'best_model.pkl')

📊 Visualization Options

# Performance comparison
automl.plot_performance(figsize=(12, 8), save_path='performance.png')

# Training time analysis
automl.plot_training_time_comparison()

# Radar chart of top 5 models
automl.plot_model_comparison_radar()

📚 Supported Models

✅ Regression (40+ Models)

LinearRegression, Ridge, Lasso, ElasticNet, BayesianRidge, KernelRidge, 
Lars, LassoLars, OrthogonalMatchingPursuit, PassiveAggressiveRegressor, 
SGDRegressor, HuberRegressor, TheilSenRegressor, RANSACRegressor, 
QuantileRegressor, DecisionTree, RandomForest, ExtraTrees, GradientBoosting, 
AdaBoost, VotingRegressor, BaggingRegressor, HistGradientBoosting, SVR, 
LinearSVR, NuSVR, KNN, MLP, GaussianProcess, XGBoost, LightGBM, CatBoost

✅ Classification (40+ Models)

LogisticRegression, GaussianNB, MultinomialNB, BernoulliNB, 
LinearDiscriminantAnalysis, QuadraticDiscriminantAnalysis, 
PassiveAggressiveClassifier, SGDClassifier, RidgeClassifier, Perceptron, 
DecisionTree, RandomForest, ExtraTrees, GradientBoosting, AdaBoost, 
VotingClassifier, BaggingClassifier, HistGradientBoosting, SVC, LinearSVC, 
NuSVC, KNN, MLP, GaussianProcess, LabelPropagation, LabelSpreading, 
XGBoost, LightGBM, CatBoost

ℹ️ Package Information

automl = AutoML()
automl.author()
╔══════════════════════════════════════════════════════════════╗
║                         AutoML Package                       ║
║    An Automated Machine Learning Library By Vinayak Bhosale  ║
╚══════════════════════════════════════════════════════════════╝

📦 Package: AutoML
👨‍💻 Author: Vinayak Bhosale
📅 Copyright: © 2025
🔖 Version: 1.0.0
📄 License: MIT License
📝 Description: A comprehensive AutoML library for automated machine learning

🚀 Features:
   🤖 Automated model selection and training
   🔄 Multi-threaded parallel processing
   📊 Comprehensive performance visualization
   🔍 Hyperparameter optimization with GridSearch
   📈 Support for regression and classification
   🎯 20+ built-in machine learning algorithms
   ⚡ Support for XGBoost, LightGBM, and CatBoost
   📋 Detailed performance metrics and comparisons
   💾 Model persistence and reusability
   🎨 Beautiful visualization with radar charts

====================================================================
Thank you for using AutoML! Happy Machine Learning! 🎉
====================================================================

📘 Documentation

For detailed usage and customization:

# View help guide
automl.help_automl()

# View method documentation
help(AutoML.fit)
help(AutoML.plot_performance)

📄 License

This project is licensed under the MIT License - see the LICENSE file for details.

Metadata

Release files for vinzy-automl 1.0.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 vinzy-automl 1.0.0
File Size Uploaded Provenance
vinzy_automl-1.0.0.tar.gz 18.6 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for vinzy-automl 1.0.0
File Interpreter ABI Platform
vinzy_automl-1.0.0-py3-none-any.whl Python 3 none any Details

Total release size: 35.3 kB

Release files / vinzy_automl-1.0.0.tar.gz

Download URL vinzy_automl-1.0.0.tar.gz
Size 18.6 kB
Tags Source
SHA-256 checksum
How to use checksums
b76cba5dd752c219811d3dc4c46b71f2d4aaf3f46e22116428d44d7354010c1a
BLAKE2b-256 checksum
How to use checksums
aafd6b3068539fe4d77918fa41a254a234ec97bf2ac5a2f7380acfcf8847c07a
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.11.5

Release files / vinzy_automl-1.0.0-py3-none-any.whl

Download URL vinzy_automl-1.0.0-py3-none-any.whl
Size 16.6 kB
Tags Python 3
SHA-256 checksum
How to use checksums
5cda15f610a37bad854a82db6f861e1334a8feca2a15d09a7afd6ac9748cb09c
BLAKE2b-256 checksum
How to use checksums
ddb932a0b17e0628bc34a0720abee7b8368ae3a7716bef2d0433df97cbbe7cb6
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.11.5

Security for vinzy-automl 1.0.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.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

1.0.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