{"templateId":"markdown","sharedDataIds":{"sidebar":"sidebar-sidebars.yaml"},"props":{"metadata":{"markdoc":{"tagList":[]},"redocly_category":"Products","product_name":"Machine Learning","type":"markdown"},"seo":{"title":"Gluon Train","description":"Treasure Data Product Documentation · Collect and Unify · Segment and Activate · Experiment and Analyze · Decisioning Automate with AI Scale and Trust.","siteUrl":"https://docs.treasuredata.com","lang":"en-US","llmstxt":{"hide":false,"sections":[{"title":"Table of contents","includeFiles":["**/*"],"excludeFiles":[]}],"excludeFiles":[]}},"dynamicMarkdocComponents":[],"compilationErrors":[],"ast":{"$$mdtype":"Tag","name":"article","attributes":{},"children":[{"$$mdtype":"Tag","name":"Heading","attributes":{"level":1,"id":"gluon-train","__idx":0},"children":["Gluon Train"]},{"$$mdtype":"Tag","name":"p","attributes":{},"children":["This notebook builds a prediction model using ",{"$$mdtype":"Tag","name":"MarkdownLink","attributes":{"href":"https://auto.gluon.ai/stable/index.html"},"children":["AutoGluon"]},", automating data processing, feature engineering, model selection, ensembling, and hyperparameter tuning."]},{"$$mdtype":"Tag","name":"p","attributes":{},"children":["AutoGluon combines several ML models using stack ensembling. Supported models include:"]},{"$$mdtype":"Tag","name":"ul","attributes":{},"children":[{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Neural network (MXNet, FastAI)"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Gradient Boosting Models (LightGBM, CatBoost, XGBoost)"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Random Forests"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Extremely Randomized Trees"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["k-Nearest Neighbors"]}]},{"$$mdtype":"Tag","name":"Heading","attributes":{"level":2,"id":"workflow-example","__idx":1},"children":["Workflow Example"]},{"$$mdtype":"Tag","name":"p","attributes":{},"children":["Find a sample workflow in ",{"$$mdtype":"Tag","name":"MarkdownLink","attributes":{"href":"https://github.com/treasure-data/treasure-boxes/blob/automl/machine-learning-box/automl/ml_experiment.dig"},"children":["Treasure Boxes"]},"."]},{"$$mdtype":"Tag","name":"CodeBlock","attributes":{"header":{"controls":{"copy":{}}},"source":"+gluon_train:\n  ml_train>:\n    notebook: gluon_train\n    model_name: gluon_model\n    input_table: ml_dataset.bank_marketing\n    target_column: loan \n    time_limit: 3 * 60 # soft time limit in seconds\n"},"children":[]},{"$$mdtype":"Tag","name":"Heading","attributes":{"level":2,"id":"parameters","__idx":2},"children":["Parameters"]},{"$$mdtype":"Tag","name":"div","attributes":{"className":"md-table-wrapper"},"children":[{"$$mdtype":"Tag","name":"table","attributes":{"className":"md"},"children":[{"$$mdtype":"Tag","name":"thead","attributes":{},"children":[{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"th","attributes":{"data-label":"Parameter name"},"children":["Parameter name"]},{"$$mdtype":"Tag","name":"th","attributes":{"data-label":"Console Name"},"children":["Console Name"]},{"$$mdtype":"Tag","name":"th","attributes":{"data-label":"Description"},"children":["Description"]},{"$$mdtype":"Tag","name":"th","attributes":{"data-label":"Default"},"children":["Default"]}]}]},{"$$mdtype":"Tag","name":"tbody","attributes":{},"children":[{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["docker.task_mem"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Docker Task Mem"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Task memory size. Available: 64g, 128g (default), 256g, 384g, 512g (tier dependent)."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["128g"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["input_table"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Input Table"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["TD table used for EDA as dbname.table_name."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["-"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["target_column"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Target Column"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Column name used for the label."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["-"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["model_name"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Model Name"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Prediction model name."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["-"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["problem_type"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Problem Type"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["One of binary, multiclass, regression, quantile. Inferred if not specified."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["None"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["oversampling_threshold"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Oversampling Threshold"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Threshold rate of minority class for SMOTE oversampling (binary only). 0 disables."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["0.001"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["proba_calibration"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Proba Calibration"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Run probability calibration after oversampling."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["True"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["eval_metric"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Eval Metric"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Automatically selected if not specified."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["None"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["ignore_columns"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Ignore Columns"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Columns to ignore."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["time"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["time_limit"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Time Limit"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Soft training limit in seconds (max 24h). Hint to AutoGluon."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["60 * 60"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["sampling_threshold"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Sampling Threshold"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Threshold used for sampling. See executed notebook."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["10_000_000"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["export_leaderboard"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Export Leaderboard"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Export leaderboard as TD table if specified."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["None"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["export_feature_importance"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Export Feature Importance"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Export feature importance as TD table if specified."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["None"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["exclude_models"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Exclude Model"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Models to ignore."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["KNN"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["hide_table_contents"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Hide Table Contents"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Suppress showing table contents."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["False"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["share_model"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Share Model"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Share trained models in an account."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["False"]}]},{"$$mdtype":"Tag","name":"tr","attributes":{},"children":[{"$$mdtype":"Tag","name":"td","attributes":{},"children":["refit_full"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Refit Full"]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["Retrain models on all data. Choices: best, false, default."]},{"$$mdtype":"Tag","name":"td","attributes":{},"children":["default"]}]}]}]}]},{"$$mdtype":"Tag","name":"p","attributes":{},"children":["Accepted eval_metric values:"]},{"$$mdtype":"Tag","name":"ul","attributes":{},"children":[{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Binary & Multiclass: accuracy, balanced_accuracy, f1, f1_macro, f1_micro, f1_weighted, average_precision, precision, precision_macro, precision_micro, precision_weighted, recall, recall_macro, recall_micro, recall_weighted, log_loss (multiclass default), pac_score"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Binary only: roc_auc (binary default), roc_auc_ovo_macro"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Regression: root_mean_squared_error (default), mean_squared_error, mean_absolute_error, median_absolute_error, r2"]},{"$$mdtype":"Tag","name":"li","attributes":{},"children":["Quantile Regression: pinball_loss (default)"]}]},{"$$mdtype":"Tag","name":"p","attributes":{},"children":["For more details, see the ",{"$$mdtype":"Tag","name":"MarkdownLink","attributes":{"href":"https://auto.gluon.ai/0.3.1/api/autogluon.predictor.md?highlight=eval_metric#module-0"},"children":["AutoGluon documentation"]},"."]}]},"headings":[{"value":"Gluon Train","id":"gluon-train","depth":1},{"value":"Workflow Example","id":"workflow-example","depth":2},{"value":"Parameters","id":"parameters","depth":2}],"frontmatter":{"seo":{"title":"Gluon Train"}},"lastModified":"2026-01-27T10:05:25.000Z","pagePropGetterError":{"message":"","name":""}},"slug":"/products/customer-data-platform/machine-learning/automl/notebook-solutions/gluon-train","userData":{"isAuthenticated":false,"teams":["anonymous"]},"isPublic":true}