安装模型修复

您可以选择几种方法来下载并开始使用 TensorFlow 模型修复

  • 要开始学习库包含的内容以及它的工作原理,请运行您感兴趣的技术的教程。请参阅 干预模型 部分,了解技术列表和教程链接。
  • 要在本地机器上下载,请安装 tensorflow-model-remediation pip 包。
  • 如果您的机器具有独特的配置,您可以从 源代码 构建您的包。

安装 pip 包

要安装 pip 包,请运行以下命令

$ pip install --upgrade tensorflow-model-remediation

此包需要 TensorFlow 2.0 或更高版本。

从源代码安装

首先,克隆 github 仓库

$ git clone https://github.com/tensorflow/model-remediation.git

从源代码构建 pip 包(如果需要,在进行任何修改后)

$ python3 setup.py sdist bdist_wheel

最后,安装您本地构建的包

$ pip install --upgrade dist/tensorflow_model_remediation-*.whl