Adición de metadatos que permiten búsquedas a las características

El código de esta sección utiliza la UpdateFeatureMetadataoperación de añadir metadatos con capacidad de búsqueda AWS SDK for Python (Boto3) a sus funciones en diferentes escenarios. Para obtener información sobre los demás idiomas para enviar una consulta, consulta Consulte también en la SageMaker APIreferencia de Amazon.

Para ver más ejemplos y recursos de Feature Store, consulteRecursos de Amazon SageMaker Feature Store.

Código de ejemplo

Tras actualizar los metadatos de una característica, puede utilizar la operación DescribeFeatureMetadata para ver las actualizaciones que ha realizado.

El siguiente código muestra un ejemplo de flujo de trabajo mediante el AWS SDK for Python (Boto3). Este ejemplo de código hace lo siguiente:

Para obtener más ejemplos y recursos de Feature Store, consulteRecursos de Amazon SageMaker Feature Store.

Paso 1: Configurar

Para empezar a usar Feature Store SageMaker, cree sesiones de Boto3 y Feature Store. Además, configure el bucket de S3 que desea utilizar para sus características. Este es su almacenamiento sin conexión. El siguiente código usa el bucket SageMaker predeterminado y le agrega un prefijo personalizado.

# SageMaker Python SDK version 2.x is required
%pip install 'sagemaker>=2.0.0'
import sagemaker
import sys

import boto3
import pandas as pd
import numpy as np
import io
from sagemaker.session import Session
from sagemaker import get_execution_role
from botocore.exceptions import ClientError


prefix = 'sagemaker-featurestore-introduction'
role = get_execution_role()

sagemaker_session = sagemaker.Session()
region = sagemaker_session.boto_region_name
s3_bucket_name = sagemaker_session.default_bucket()
sagemaker_client = boto_session.client(service_name='sagemaker', region_name=region)
            

Paso 2: Crear un grupo de características y agregar características

El siguiente código es un ejemplo de creación de un grupo de características con definiciones de características.

feature_group_name = "test-for-feature-metadata"
feature_definitions = [
    {"FeatureName": "feature-1", "FeatureType": "String"},
    {"FeatureName": "feature-2", "FeatureType": "String"},
    {"FeatureName": "feature-3", "FeatureType": "String"},
    {"FeatureName": "feature-4", "FeatureType": "String"},
    {"FeatureName": "feature-5", "FeatureType": "String"}
]
try:
    sagemaker_client.create_feature_group(
        FeatureGroupName=feature_group_name,
        RecordIdentifierFeatureName="feature-1",
        EventTimeFeatureName="feature-2",
        FeatureDefinitions=feature_definitions,
        OnlineStoreConfig={"EnableOnlineStore": True}
    )
except ClientError as e:
    if e.response["Error"]["Code"] == "ResourceInUse":
        pass
    else:
        raise e       
     

Paso 3: Agregar metadatos

Antes de agregar metadatos, utilice la operación DescribeFeatureGroup para asegurarse de que el estado del grupo de características sea Created.

sagemaker_client.describe_feature_group(
        FeatureGroupName=feature_group_name
    )               
            

Agregue una descripción a la característica.

sagemaker_client.update_feature_metadata(
    FeatureGroupName=feature_group_name,
    FeatureName="feature-1",
    Description="new description"
)
            

Puede utilizar la DescribeFeatureMetadataoperación para comprobar si ha actualizado correctamente la descripción del grupo de elementos.

    sagemaker_client.describe_feature_metadata(
    FeatureGroupName=feature_group_name,
    FeatureName="feature-1"
)

También puede utilizarla para agregar parámetros al grupo de características.

sagemaker_client.update_feature_metadata(
    FeatureGroupName=feature_group_name,
    FeatureName="feature-1",
    ParameterAdditions=[
        {"Key": "team", "Value": "featurestore"},
        {"Key": "org", "Value": "sagemaker"},
    ]
)
            

Puede volver a usar la operación DescribeFeatureMetadata para comprobar si ha agregado los parámetros correctamente.

    sagemaker_client.describe_feature_metadata(
    FeatureGroupName=feature_group_name,
    FeatureName="feature-1"
)