You can deploy this model and start earning money today!

Iris314/
classical-automl-model

Name: classical-automl-model
Availability: InStock

automl

tabular-classification

Views

No views yet

Model Card for Lego Brick Classification (Classical AutoML)

This model classifies LEGO pieces into three types — Standard, Flat, and Sloped — using their geometric dimensions (Length, Height, Width, Studs).
It was trained using AutoGluon Tabular AutoML, which automatically searched over classical ML models (LightGBM, XGBoost, CatBoost, Random Forest, k-NN, Neural Network) and selected the best-performing one.

Model Details

Model Description

Developed by: Xinxuan Tang (CMU)
Dataset curated by: Anuhya Edupuganti (CMU)
Model type: AutoML ensemble (best model = LightGBM)
Language(s): N/A (tabular data)
Finetuned from: Not applicable

Model Sources

Repository: Hugging Face Model Repo
Dataset: aedupuga/lego-sizes

Uses

Direct Use

Educational practice in tabular classification.
Experimenting with AutoML search and hyperparameter tuning.

Downstream Use

Could be used as a teaching example for AutoML pipelines on small tabular datasets.

Out-of-Scope Use

Not suitable for industrial LEGO quality control, since dataset is synthetic and small.

Bias, Risks, and Limitations

Small dataset: only 30 original bricks, augmented to 300 synthetic samples.
Synthetic data bias: jitter augmentation may not reflect real-world LEGO variations.

Recommendations

Users should treat results as proof-of-concept and not deploy in production.

How to Get Started with the Model

python

from autogluon.tabular import TabularPredictor
import pandas as pd

# Load trained predictor
predictor = TabularPredictor.load("autogluon_model/")

# Run inference
test_data = pd.DataFrame([{"Length": 4, "Height": 1.2, "Width": 2, "Studs": 4}])
print(predictor.predict(test_data))

Iris314/classical-automl-model