Training data requirements for Clean Rooms ML - AWS Clean Rooms

Training data requirements for Clean Rooms ML

To successfully create a lookalike model, your training data must meet the following requirements:

  • The training data must be in Parquet, CSV, or JSON format.

  • Your training data must be cataloged in AWS Glue. For more information, see Getting started with the AWS Glue Data Catalog in the AWS Glue Developer Guide. We recommend using AWS Glue crawlers to create your tables because the schema is inferred automatically.

  • The Amazon S3 bucket that contains the training data and seed data is in the same AWS region as your other Clean Rooms ML resources.

  • The training data must contain at least 100,000 unique user IDs with at least two item interactions each.

  • The training data must contain at least 1 million records.

  • The schema specified in the CreateTrainingDataset action must align with the schema defined when the AWS Glue table was created.

  • The required fields, as defined in the provided table, are defined in the CreateTrainingDataset action.

    Field type Supported data types Required Description
    USER_ID string, int, bigint Yes A unique identifier for each user in the dataset. It should be a non-Personally Identifiable Information (PII) value. This might be a hashed identifier or a customer ID.
    ITEM_ID string, int, bigint Yes A unique identifier for each item a user interacts with.
    TIMESTAMP bigint, int, timestamp Yes The time when a user interacted with the item. Values must be in the Unix epoch time in seconds format.
    CATEGORICAL_FEATURE string, int, float, bigint, double, boolean, array No Captures categorical data related to the user or the item. This can include things like an event type (such as click or purchase), user demographics (age group, gender - anonymized), user location (city, country - anonymized), item category (such as clothing or electronics), or item brand.
    NUMERICAL_FEATURE double, float, int, bigint No Captures numerical data related to the user or the item. This can include things like user purchase history (total amount spent), item price, number of times an item is visited, or user ratings for items.

  • Optionally, you can provide up to 10 total categorical or numerical features.

Here is an example of a valid training data set in CSV format

USER_ID,ITEM_ID,TIMESTAMP,EVENT_TYPE(CATEGORICAL FEATURE),EVENT_VALUE (NUMERICAL FEATURE)
196,242,881250949,click,15
186,302,891717742,click,13
22,377,878887116,click,10
244,51,880606923,click,20
166,346,886397596,click,10