TabularFoundationModel

class autogluon.cloud.TabularFoundationModel(model_id: str, **kwargs)

Foundation model for tabular prediction on Amazon SageMaker.

Wraps pretrained tabular models like Mitra, with no training required. Each model_id targets a single task, either classification (*-classifier) or regression (*-regressor). Use list_models() to get the supported model_id values.

Predictions can be produced in a one-off SageMaker job with predict() / predict_proba(), or through a real-time endpoint created with deploy(). In both modes, labeled train_data provides the in-context examples for each prediction.

Parameters:

• model_id (str) – ID of the foundation model, e.g. "chronos-2" or "mitra-classifier". Use list_models() to get the supported IDs.

• cloud_output_path (str | None, default = None) –

  S3 location where intermediate artifacts are stored. Accepts:

  • s3://bucket — a unique timestamped subfolder ag-<timestamp> is appended.

  • s3://bucket/prefix — used verbatim. Re-running with the same prefix will overwrite previously written artifacts.

  • None (default) — use the bucket saved in ~/.autogluon/cloud.yaml (set by autogluon.cloud.bootstrap() / autogluon.cloud.register()) and append a timestamped subfolder. Raises if no bucket is configured.

• role (str | None, default = None) – ARN of the SageMaker execution role used to run training and inference jobs. If None, falls back to role_arn in ~/.autogluon/cloud.yaml (set by autogluon.cloud.bootstrap() / autogluon.cloud.register()), and finally to the role of the current AWS identity.

• hyperparameters (dict[str, Any] | None, default = None) – Model hyperparameters. Hyperparameters passed to predict() or deploy() take precedence.

• model_artifact_uri (str | None, default = None) – S3 URI of a model.tar.gz created by cache_model_artifact(). If set, deploy() loads the weights from this artifact instead of downloading them from Hugging Face.

• backend (Literal["sagemaker"], default = "sagemaker") – Cloud backend to use.

Methods

cache_model_artifact	Bundle the model weights from Hugging Face and the serve script into a model.tar.gz on S3.
deploy	Deploy the model to a real-time endpoint.
from_dict	Create a model from the output of to_dict().
from_json	Create a model from the output of to_json().
list_models	List the supported model_id values.
predict	Predict test_data in a one-off SageMaker job, using the labeled train_data as in-context examples.
predict_proba	Predict class probabilities for test_data in a one-off SageMaker job, using the labeled train_data as in-context examples.
to_dict	Serialize model_id, hyperparameters, and model_artifact_uri to a dict.
to_json	Serialize model_id, hyperparameters, and model_artifact_uri to a JSON string.