Batch write and discover records in Amazon SageMaker Feature Store
Amazon SageMaker Feature Store now supports two new APIs: BatchWriteRecord writes up to 25 records across multiple feature groups in a single call, and ListRecords enumerates record identifiers within a feature group. In this post, we walk through each API with code examples you can use to get started.
Amazon SageMaker Feature Store now offers two new APIs to simplify feature ingestion: BatchWriteRecord and ListRecords. BatchWriteRecord allows writing up to 25 feature records across multiple feature groups in a single API call, addressing the throughput limitations of the existing single-record PutRecord API. The API supports both online and offline storage tiers, with partial-success semantics ensuring individual record failures do not halt the entire operation.
ListRecords enables enumeration of record identifiers within a feature group, providing a way to discover existing records in both Standard (DynamoDB-backed) and In-Memory (Redis-backed) storage tiers. These new APIs aim to streamline feature pipeline management for high-throughput ML workflows, reducing connection overhead and enabling more efficient data retrieval.
Written by urgent.news from AWS Machine Learning's reporting — not their text. Machine-written — may contain errors; check the original before relying on it.
This story
This is one outlet's version. Read the fullest account.
