TensorflowCatalog 类

  • 参考

定义

命名空间:
Microsoft.ML
程序集:
Microsoft.ML.TensorFlow.dll
包:
Microsoft.ML.TensorFlow v3.0.1
包:
Microsoft.ML.TensorFlow v1.2.0
包:
Microsoft.ML.TensorFlow v1.3.1
包:
Microsoft.ML.TensorFlow v1.4.0
包:
Microsoft.ML.TensorFlow v1.5.5
包:
Microsoft.ML.TensorFlow v1.6.0
包:
Microsoft.ML.TensorFlow v1.7.0
包:
Microsoft.ML.TensorFlow v2.0.0

这 TensorFlowTransformer 两种方案在以下两种情况下使用。

  1. 使用预先训练的 TensorFlow 模型评分：在此模式下，转换从预先训练的 Tensorflow 模型中提取隐藏层的值，并使用输出作为 ML.Net 管道中的特征。
  2. 重新训练 TensorFlow 模型：在此模式下，转换使用通过 ML.Net 管道传递的用户数据重新训练 TensorFlow 模型。 训练模型后，输出可用作评分功能。 
public static class TensorflowCatalog
type TensorflowCatalog = class
Public Module TensorflowCatalog
继承
Object
TensorflowCatalog

注解

TensorFlowTransform 使用预先训练的 Tensorflow 模型提取指定的输出。 （可选）它可以进一步重新训练用户数据的 TensorFlow 模型，以调整用户数据上的模型参数 (也称为“传输学习”) 。

为了评分，转换采用预先训练的 Tensorflow 模型输入、输入节点的名称以及要提取其值输出节点的名称。 为了重新训练，转换还需要训练相关的参数，例如 TensorFlow 图中的优化操作名称、图形中的学习速率操作的名称及其值、用于计算损失和性能指标的操作的名称等。

此转换要求安装 Microsoft.ML.TensorFlow nuget。 TensorFlowTransform 对输入、输出、数据处理和重新训练有以下假设。

  1. 对于输入模型，当前 TensorFlowTransform 支持 冻结模型 格式和 SavedModel 格式。 但是，只能对 SavedModel 格式重新训练模型。 检查点 格式目前既不支持评分，也不支持重新训练，因为无法加载 TensorFlow C-API。
  2. 转换一次仅支持对一个示例进行评分。 但是，可以分批执行重新训练。
  3. 高级传输学习/微调方案 (例如，在网络中添加更多层、更改输入形状、冻结在重新训练过程中不需要更新的层等。) 目前无法实现，因为目前无法使用 TensorFlow C-API 在模型中支持网络/图形操作。
  4. 输入列的名称 (s) 应与 TensorFlow 模型中输入 (s) 的名称匹配。
  5. 每个输出列的名称应与 TensorFlow 图中的一个操作匹配。
  6. 目前，double、float、long、int、short、sbyte、ulong、uint、ushort、byte 和 bool 是输入/输出的可接受数据类型。
  7. 成功后，转换将引入对应于指定的每个输出列的新列 IDataView 。

可以使用或summarize_graph工具获取 GetModelSchema() TensorFlow 模型的输入和输出。

方法

LoadTensorFlowModel(ModelOperationsCatalog, String)

将 TensorFlow 模型加载到内存中。 这是一种方便的方法，允许加载模型一次，然后使用它来查询架构和使用TensorFlowEstimatorScoreTensorFlowModel(String, String, Boolean)。 使用此 API 需要 TensorFlow redist 上的其他 NuGet 依赖项，有关详细信息，请参阅链接文档。 TensorFlowModel 还保留对非托管资源的引用，这些资源需要通过显式调用 Dispose () 或隐式声明带有“using”语法的变量/>
LoadTensorFlowModel(ModelOperationsCatalog, String, Boolean)

将 TensorFlow 模型加载到内存中。 这是一种方便的方法，允许加载模型一次，然后使用它来查询架构和使用TensorFlowEstimatorScoreTensorFlowModel(String, String, Boolean)。 使用此 API 需要 TensorFlow redist 上的其他 NuGet 依赖项，有关详细信息，请参阅链接文档。 TensorFlowModel 还保留对非托管资源的引用，这些资源需要通过显式调用 Dispose () 或隐式声明带有“using”语法的变量/>

适用于

另请参阅