ml-auto-trainer 0.1.0

A simple auto machine learning trainer that builds models with one function call.

ml_auto_trainer 🚀

A one-function AutoML package that builds, trains, and evaluates ML models automatically.

✨ Features

• Automatically detects problem type (classification or regression)
• Supports popular models (RandomForest, XGBoost, Logistic, Linear, KNN, GBM)
• Prints metrics automatically

🧠 Usage

from ml_auto_trainer.auto_trainer import auto_train
import pandas as pd
from sklearn.datasets import load_iris

# Load data
data = load_iris()
X = pd.DataFrame(data.data, columns=data.feature_names)
y = data.target

# Train model
model = auto_train(X, y, model_name='xgboost')