Prepare data
Upload CSV or Excel data, inspect columns, set a target, and save preprocessing choices.
AutoTrain ML turns a dataset-to-model workflow into a guided interface while keeping model settings, metrics, and artifacts visible.
AutoTrain keeps preparation, training, comparison, testing, and export attached to the same project so the experiment can be reopened from another device without rebuilding the setup.
Upload CSV or Excel data, inspect columns, set a target, and save preprocessing choices.
Choose the task, algorithm, validation settings, and hyperparameters instead of handing every decision to a black box.
Compare runs, test new rows, download results, and expose supported models through prediction endpoints.
Create an account and upload a clean tabular dataset.
Choose classification, regression, or clustering and configure preprocessing.
Train one or more algorithms and compare the evaluation output.
Create a testing tab, validate new data, and export the selected model.
Choose the task from the problem definition. AutoTrain does not silently change the method or hyperparameters selected by the user.
| Task | Typical target | Evaluation focus | Testing output |
|---|---|---|---|
| Classification | Category or class | Accuracy, precision, recall, F1, confusion matrix, ROC/PR where available | Predicted class, confidence, probabilities |
| Regression | Continuous numeric value | MAE, RMSE, R², residual behavior | Predicted numeric value |
| Clustering | No supervised target | Silhouette, cluster size, stability, interpretability | Assigned cluster and distance context |
Dataset reference, preprocessing settings, task, algorithm, hyperparameters, validation metrics, and generated visualizations remain attached to the saved experiment.
Use custom single-row values or a schema-matched CSV to test new data. Results can be exported as CSV for downstream review.
Completed models can be exported with the preprocessing and input schema required to reproduce predictions correctly.
Supported models expose a schema endpoint and a prediction endpoint that can be called from an application backend.
AutoML research describes automation across data preparation, model selection, optimization, and evaluation. AutoTrain applies that principle selectively: repetitive execution is automated while task, preprocessing, algorithm, and parameter choices remain visible.
He, X., Zhao, K., & Chu, X. (2021). AutoML: A survey of the state-of-the-art. Knowledge-Based Systems, 212, 106622. Read the journal article
No. It reports comparable experiment results; the user chooses methods, parameters, and the final model based on the actual objective.
Yes. Saved datasets, experiments, model metadata, visual results, and testing tabs are synchronized through AutoTrain Engine.
No. Benchmarks are fixed reference runs. Real performance depends on data quality, the split strategy, leakage, drift, and the cost of each error type.