AWS SageMaker

AWS SageMaker

Alpha release

This source is currently in alpha. The interface and available tables may change.

Let AI connect your sources for you

Skip the manual setup — run this in your project and the wizard auto-detects your databases and APIs and connects them to PostHog.

Learn more
PostHog Wizard hedgehog

Connect AWS SageMaker to PostHog to sync your data into the PostHog data warehouse for analysis and modeling.

Sync SageMaker resources from one AWS region. Grant these IAM permissions for the tables you select: sagemaker:ListTrainingJobs, sagemaker:DescribeTrainingJob, sagemaker:ListProcessingJobs, sagemaker:DescribeProcessingJob, sagemaker:ListEndpoints, sagemaker:DescribeEndpoint, sagemaker:ListModels, sagemaker:DescribeModel, sagemaker:ListEndpointConfigs, sagemaker:DescribeEndpointConfig, sagemaker:ListModelPackages, sagemaker:DescribeModelPackage, sagemaker:ListHyperParameterTuningJobs, sagemaker:DescribeHyperParameterTuningJob, sagemaker:ListAutoMLJobs, and sagemaker:DescribeAutoMLJobV2. Jobs, endpoints, and model packages use full refresh to capture status changes. Models and endpoint configurations also support incremental sync by creation time.

Configuration

OptionDescription
AWS access key ID
Type: text
Required: True
AWS secret access key
Type: password
Required: True
AWS session token
Type: password
Required: False

Temporary credentials require a session token. Scheduled syncs fail when the token expires.

AWS region
Type: text
Required: True

Linking AWS SageMaker to PostHog

  1. Go to the Data pipeline page in PostHog
  2. Click New source and select AWS SageMaker
  3. Fill in the required configuration fields
  4. Click Next, select the tables you want to sync, and then press Import

Supported tables

TableDescriptionSync methodIncremental fieldPrimary key
training_jobs

Training jobs with status, resource settings, metrics, and billable duration.

Full refresh——
processing_jobs

Processing jobs with status, input and output settings, and compute resources.

Full refresh——
endpoints

Inference endpoints with status, deployment settings, and production variants.

Full refresh——
models

Models with container settings, execution roles, and creation times.

Incremental, Full refreshcreation_time—
endpoint_configs

Endpoint configurations with production variants, instance types, and data capture settings.

Incremental, Full refreshcreation_time—
model_packages

Model packages and Model Registry versions with approval status, metrics, and inference settings.

Full refresh——
hyperparameter_tuning_jobs

Hyperparameter tuning jobs with objectives, best training jobs, and consumed resources.

Full refresh——
automl_jobs

AutoML jobs with problem type, objective, status, and best candidate.

Full refresh——