Transform your data into production-ready ML models in minutes, not hours!
π― What if I told you that you could build a complete machine learning pipeline with just 5 lines of code?
AutoML Lite is here to revolutionize how you approach machine learning projects. Whether you're a data scientist, ML engineer, or just getting started with AI, this library will save you countless hours of boilerplate code and configuration headaches.
π¬ See It In Action
AutoML Lite Complete Pipeline Demo
Generated Interactive HTML Reports
Weights & Biases Integration
π€ Why AutoML Lite?
The Problem with Traditional ML Workflows
- Hours of boilerplate code for data preprocessing
- Manual hyperparameter tuning that takes forever
- Complex configuration management across different projects
- No standardized way to track experiments
- Limited interpretability out of the box
- Difficult deployment and model management
The AutoML Lite Solution
β
Zero Configuration Required - Works out of the box
β
Complete ML Pipeline - From data to deployment
β
Production Ready - Built for real-world applications
β
Advanced Features - Deep learning, time series, interpretability
β
Experiment Tracking - MLflow, W&B, TensorBoard integration
β
Interactive Reports - Beautiful HTML reports with visualizations
π Installation & Quick Start
Install in 30 seconds
pip install automl-lite
Build Your First Model in 5 Lines
from automl_lite import AutoMLite
import pandas as pd
# Load your data
data = pd.read_csv('your_data.csv')
# Initialize AutoML (that's it!)
automl = AutoMLite(time_budget=300)
# Train and get the best model
best_model = automl.fit(data, target_column='target')
# Make predictions
predictions = automl.predict(new_data)
That's it! Your model is trained, optimized, and ready for production. No more endless configuration files or manual tuning!
π― What Makes AutoML Lite Special?
π§ Intelligent Automation
- Auto Feature Engineering: Automatically creates 232 features from 20 original features (11.6x expansion!)
- Smart Model Selection: Tests 15+ algorithms and picks the best one
- Hyperparameter Optimization: Uses Optuna for efficient tuning
- Ensemble Methods: Automatically creates voting classifiers for better performance
π Production-Ready Features
- Deep Learning Support: TensorFlow and PyTorch integration
- Time Series Forecasting: ARIMA, Prophet, and LSTM models
- Advanced Interpretability: SHAP, LIME, permutation importance
- Experiment Tracking: MLflow, Weights & Biases, TensorBoard
- Interactive Dashboards: Real-time monitoring with Streamlit
π Comprehensive Reporting
- Interactive HTML Reports: Beautiful visualizations with Plotly
- Model Performance Analysis: Confusion matrices, ROC curves, residuals
- Feature Importance: Detailed feature analysis and correlation matrices
- Training History: Complete training logs and performance metrics
π₯ Real-World Performance
Test Results (Production Demo)
- Training Time: 391.92 seconds for complete pipeline
- Best Model: Random Forest (80.00% accuracy)
- Feature Engineering: 20 β 232 features (11.6x expansion)
- Feature Selection: 132/166 features intelligently selected
- Hyperparameter Optimization: 50 trials with Optuna
Supported Problem Types
- β Classification (Binary & Multi-class)
- β Regression
- β Time Series Forecasting
- β Deep Learning Tasks
π οΈ Advanced Usage Examples
Custom Configuration
from automl_lite import AutoMLite
# Custom configuration
config = {
'time_budget': 600,
'max_models': 20,
'cv_folds': 5,
'feature_engineering': True,
'ensemble_method': 'voting',
'interpretability': True
}
automl = AutoMLite(**config)
Time Series Forecasting
# Time series data
automl = AutoMLite(problem_type='time_series')
model = automl.fit(data, target_column='sales', date_column='date')
# Get forecasts
forecast = automl.predict_future(periods=30)
Deep Learning
# Deep learning with TensorFlow
automl = AutoMLite(
include_deep_learning=True,
deep_learning_framework='tensorflow'
)
model = automl.fit(data, target_column='target')
π CLI Interface
Command Line Usage
# Basic usage
automl-lite train data.csv --target target_column
# With custom config
automl-lite train data.csv --target target_column --config config.yaml
# Generate report
automl-lite report --model model.pkl --output report.html
π¨ Interactive Dashboard
Launch the interactive dashboard for real-time monitoring:
from automl_lite.ui import launch_dashboard
# Launch dashboard
launch_dashboard(automl)
π Model Interpretability
AutoML Lite provides comprehensive model interpretability:
# Get SHAP values
shap_values = automl.explain_model(X_test)
# Feature importance
importance = automl.get_feature_importance()
# Partial dependence plots
automl.plot_partial_dependence('feature_name')
π¦ Installation Options
From PyPI (Recommended)
pip install automl-lite
From Source
git clone https://github.com/Sherin-SEF-AI/AutoML-Lite.git
cd AutoML-Lite
pip install -e .
π― Use Cases
Perfect For:
- π’ Data Scientists - Rapid prototyping and experimentation
- π ML Engineers - Production model development
- π Analysts - Quick insights from data
- π Students - Learning machine learning concepts
- π Startups - Fast MVP development
Industries:
- Finance: Credit scoring, fraud detection
- Healthcare: Disease prediction, patient monitoring
- E-commerce: Customer segmentation, demand forecasting
- Marketing: Campaign optimization, customer lifetime value
- Manufacturing: Predictive maintenance, quality control
π Getting Started Guide
1. Install the Package
pip install automl-lite
2. Prepare Your Data
import pandas as pd
# Load your dataset
data = pd.read_csv('your_dataset.csv')
# Ensure your target column is present
print(data.columns)
3. Run AutoML
from automl_lite import AutoMLite
# Initialize with default settings
automl = AutoMLite()
# Train your model
best_model = automl.fit(data, target_column='your_target_column')
# Make predictions
predictions = automl.predict(new_data)
4. Generate Report
# Generate comprehensive HTML report
automl.generate_report('my_ml_report.html')
π§ Configuration Templates
AutoML Lite comes with pre-built configuration templates:
- Basic: Quick experiments and prototyping
- Production: Optimized for production deployment
- Research: Extensive hyperparameter search
- Customer Churn: Specialized for churn prediction
- Fraud Detection: Optimized for fraud detection tasks
- House Price: Specialized for real estate prediction
π€ Contributing
We welcome contributions! Here's how you can help:
- Fork the repository
- Create a feature branch
- Make your changes
- Add tests
- Submit a pull request
Development Setup
git clone https://github.com/Sherin-SEF-AI/AutoML-Lite.git
cd AutoML-Lite
pip install -r requirements.txt
pip install -e .
π Documentation & Resources
- π Full Documentation: GitHub Wiki
- π― API Reference: API Docs
- π Examples: Example Notebooks
- π Quick Start: Installation Guide
π What's Next?
Upcoming Features
- π AutoML Pipeline Versioning
- π Cloud Deployment Integration
- π± Mobile Model Optimization
- π Privacy-Preserving ML
- π Multi-Language Support
π¬ Join the Community
- π Star the Repository: GitHub
- π Report Issues: Issue Tracker
- π‘ Feature Requests: Discussions
- π§ Contact: sherin.joseph2217@gmail.com
π Why Choose AutoML Lite?
Feature | AutoML Lite | Other Libraries |
---|---|---|
Setup Time | 30 seconds | 30+ minutes |
Configuration | Zero required | Complex configs |
Production Ready | β Built-in | β Manual setup |
Deep Learning | β Integrated | β Separate setup |
Time Series | β Native support | β Limited |
Interpretability | β Advanced | β Basic |
Experiment Tracking | β Multi-platform | β Limited |
Interactive Reports | β Beautiful HTML | β Basic plots |
π― Ready to Transform Your ML Workflow?
Stop spending hours on boilerplate code. Start building amazing ML models in minutes!
pip install automl-lite
Top comments (0)