import sagemaker
from sagemaker.predictor import Predictor
from sagemaker.serializers import JSONSerializer
from sagemaker.deserializers import JSONDeserializer
# Specify your endpoint name
endpoint_name = "your_endpoint_name"
predictor = Predictor(endpoint_name=endpoint_name,
serializer=JSONSerializer(),
deserializer=JSONDeserializer())
# Test sample
sample = {"text": "example text to predict"}
# Invoke endpoint predictor
prediction_response = predictor.predict(sample)
# View prediction output
print(prediction_response)
如果预测输出看起来正确,那么模型应该可以正确分类测试数据集中的样本。