匯入 SageMaker AI 模型

您可以選擇將 SageMaker AI 託管模型匯入至 Amazon Fraud Detector。與模型類似，SageMaker AI 模型可以新增至偵測器，並使用 GetEventPrediction API 產生詐騙預測。在GetEventPrediction請求中，Amazon Fraud Detector 將調用您的 SageMaker AI 端點，並將結果傳遞給您的規則。

您可以設定 Amazon Fraud Detector 使用作為GetEventPrediction請求一部分傳送的事件變數。如果您選擇使用事件變數，則必須提供輸入範本。Amazon Fraud Detector 將使用此範本將您的事件變數轉換為必要的輸入承載，以叫用 SageMaker AI 端點。或者，您可以將 SageMaker AI 模型設定為使用作為GetEventPrediction請求一部分傳送的 byteBuffer。

Amazon Fraud Detector 支援匯入使用 JSON 或 CSV 輸入格式和 JSON 或 CSV 輸出格式的 SageMaker AI 演算法。支援的 SageMaker AI 演算法範例包括 XGBoost、線性學習器和隨機切割森林。

使用 匯入 SageMaker AI 模型 適用於 Python (Boto3) 的 AWS SDK

若要匯入 SageMaker AI 模型，請使用 PutExternalModel API。下列範例假設 SageMaker AI 端點sagemaker-transaction-model已部署、為 InService 狀態，並使用 XGBoost 演算法。

輸入組態指定 將使用事件變數來建構模型輸入 (useEventVariables 設定為 TRUE)。輸入格式為 TEXT_CSV，因為 XGBoost 需要 CSV 輸入。csvInputTemplate 指定如何從作為GetEventPrediction請求一部分傳送的變數建構 CSV 輸入。此範例假設您已建立變數 order_amt、 prev_amthist_amt和 payment_type。

輸出組態會指定 SageMaker AI 模型的回應格式，並將適當的 CSV 索引映射至 Amazon Fraud Detector 變數 sagemaker_output_score。設定完成後，您可以在規則中使用輸出變數。

注意

來自 SageMaker AI 模型的輸出必須映射至具有來源 的變數EXTERNAL_MODEL_SCORE。您無法在主控台中使用變數建立這些變數。您必須在設定模型匯入時建立它們。

import boto3
fraudDetector = boto3.client('frauddetector')

fraudDetector.put_external_model (
modelSource = 'SAGEMAKER',
modelEndpoint = 'sagemaker-transaction-model',
invokeModelEndpointRoleArn = 'your_SagemakerExecutionRole_arn',
inputConfiguration = {
    'useEventVariables' : True,
    'eventTypeName' : 'sample_transaction',
    'format' : 'TEXT_CSV',
    'csvInputTemplate' : '{{order_amt}}, {{prev_amt}}, {{hist_amt}}, {{payment_type}}'
},

outputConfiguration = {
    'format' : 'TEXT_CSV',
    'csvIndexToVariableMap' : {
        '0' : 'sagemaker_output_score'
    }
},
    
modelEndpointStatus = 'ASSOCIATED'
)