文本分类 – TensorFlow - Amazon SageMaker AI
文本分类 – TensorFlow 算法的 Amazon EC2 实例推荐功能示例笔记本

文本分类 – TensorFlow

Amazon SageMaker AI 文本分类 – TensorFlow 算法是一种有监督学习算法，支持使用 TensorFlow Hub 中的多种预先训练的模型进行迁移学习。使用迁移学习，即使没有大量文本数据可用，也可以在您自己的数据集上对一个可用的预先训练模型进行微调。文本分类算法将文本字符串作为输入，并输出每个类标签的概率。训练数据集必须为 CSV 格式。本页包含有关 Amazon EC2 实例推荐功能和文本分类 - TensorFlow 的示例笔记本的信息。

主题

文本分类 – TensorFlow 算法的 Amazon EC2 实例推荐功能

文本分类 – TensorFlow 算法支持使用所有 CPU 和 GPU 实例进行训练，包括：

  • ml.p2.xlarge

  • ml.p2.16xlarge

  • ml.p3.2xlarge

  • ml.p3.16xlarge

  • ml.g4dn.xlarge

  • ml.g4dn.16.xlarge

  • ml.g5.xlarge

  • ml.g5.48xlarge

对于大批量训练，建议使用具有更多内存的 GPU 实例。CPU（例如 M5）实例和 GPU（P2、P3、G4dn 或 G5）实例都可用于推理。有关各 AWS 区域的 SageMaker 训练和推理实例的完整列表，请参阅 Amazon SageMaker 定价。

文本分类 – TensorFlow 示例笔记本

有关如何使用 SageMaker AI 文本分类 – TensorFlow 算法在自定义数据集上进行迁移学习的更多信息，请参阅 JumpStart 简介 – 文本分类笔记本。

有关如何创建和访问可用于在 SageMaker AI 中运行示例的 Jupyter Notebook 实例的说明，请参阅 Amazon SageMaker Notebook 实例。创建笔记本实例并将其打开后，选择 SageMaker AI 示例选项卡以查看所有 SageMaker AI 示例的列表。要打开笔记本，请选择其使用选项卡，然后选择创建副本。