Hinzufügen durchsuchbarer Metadaten zu Ihren Funktionen

Der Code in diesem Abschnitt verwendet den UpdateFeatureMetadataVorgang in, AWS SDK for Python (Boto3) um Ihren Features durchsuchbare Metadaten für verschiedene Szenarien hinzuzufügen. Informationen zu den anderen Sprachen, in denen Sie eine Anfrage einreichen können, finden Sie unter „Siehe auch“ in der SageMaker APIAmazon-Referenz.

Weitere Beispiele und Ressourcen für den Feature Store finden Sie unterRessourcen für den Amazon SageMaker Feature Store.

Beispiel-Code

Nachdem Sie die Metadaten für ein Feature aktualisiert haben, können Sie den DescribeFeatureMetadata Vorgang verwenden, um die von Ihnen vorgenommenen Aktualisierungen anzuzeigen.

Der folgende Code durchläuft einen Beispiel-Workflow mit dem AWS SDK for Python (Boto3). Das Codebeispiel führt die folgenden Aufgaben durch:

Weitere Beispiele und Ressourcen für den Feature Store finden Sie unterRessourcen für den Amazon SageMaker Feature Store.

Schritt 1: Einrichtung

Um mit der Verwendung von Feature Store zu beginnen SageMaker, erstellen Sie Boto3- und Feature Store-Sitzungen. Richten Sie dann den S3-Bucket ein, den Sie für Ihre Funktionen verwenden möchten. Dies ist Ihr Offline-Speicher. Der folgende Code verwendet den SageMaker Standard-Bucket und fügt ihm ein benutzerdefiniertes Präfix hinzu.

# 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)
            

Schritt 2: Erstellen einer Feature-Gruppe und Hinzufügen von Funktionen

Der folgende Code ist ein Beispiel für die Erstellung einer Feature-Gruppe mit Feature-Definitionen.

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       
     

Schritt 3: Hinzufügen von Metadaten

Stellen Sie vor dem Hinzufügen von Metadaten mithilfe des DescribeFeatureGroup Vorgangs sicher, dass der Status der Feature-Gruppe Created lautet.

sagemaker_client.describe_feature_group(
        FeatureGroupName=feature_group_name
    )               
            

Fügen Sie dem Feature eine Beschreibung hinzu.

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

Sie können den DescribeFeatureMetadataVorgang verwenden, um festzustellen, ob Sie die Beschreibung für die Feature-Gruppe erfolgreich aktualisiert haben.

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

Sie können ihn auch verwenden, um der Featuregruppe Parameter hinzuzufügen.

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

Sie können den DescribeFeatureMetadata Vorgang erneut verwenden, um zu überprüfen, ob Sie die Parameter erfolgreich hinzugefügt haben.

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