Análisis del modelo de TensorFlow

TensorFlow Model Analysis (TFMA) es una biblioteca para evaluar modelos de TensorFlow. Permite a los usuarios evaluar sus modelos sobre grandes cantidades de datos de forma distribuida, utilizando las mismas métricas definidas en su entrenador. Estas métricas se pueden calcular en diferentes segmentos de datos y visualizarse en cuadernos de Jupyter.

Instalación

La forma recomendada de instalar TFMA es usando el paquete PyPI :

pip install tensorflow-model-analysis

Construir TFMA desde la fuente

Para compilar desde la fuente, siga los siguientes pasos:

Instale el protoc según el enlace mencionado: protoc

Cree un entorno virtual ejecutando los comandos

python3 -m venv <virtualenv_name>
source <virtualenv_name>/bin/activate
pip3 install setuptools wheel
git clone https://github.com/tensorflow/model-analysis.git
cd model-analysis
python3 setup.py bdist_wheel

Esto construirá la rueda TFMA en el directorio dist. Para instalar la rueda desde el directorio dist, ejecute los comandos

cd dist
pip3 install tensorflow_model_analysis-<version>-py3-none-any.whl

Paquetes Nocturnos

TFMA también aloja paquetes nocturnos en https://pypi-nightly.tensorflow.org en Google Cloud. Para instalar el último paquete nocturno, utilice el siguiente comando:

pip install -i https://pypi-nightly.tensorflow.org/simple tensorflow-model-analysis

Esto instalará los paquetes nocturnos para las principales dependencias de TFMA, como TensorFlow Metadata (TFMD), TFX Basic Shared Libraries (TFX-BSL).

Actualmente, TFMA requiere que TensorFlow esté instalado pero no tiene una dependencia explícita en el paquete TensorFlow PyPI. Consulte las guías de instalación de TensorFlow para obtener instrucciones.

Para habilitar la visualización de TFMA en Jupyter Notebook:

  jupyter nbextension enable --py widgetsnbextension
  jupyter nbextension enable --py tensorflow_model_analysis

Laboratorio Jupyter

Al momento de escribir, debido a https://github.com/pypa/pip/issues/9187, es posible que pip install nunca finalice. En ese caso, debe revertir pip a la versión 19 en lugar de 20: pip install "pip<20" .

El uso de una extensión de JupyterLab requiere la instalación de dependencias en la línea de comandos. Puede hacerlo dentro de la consola en la interfaz de usuario de JupyterLab o en la línea de comando. Esto incluye la instalación por separado de las dependencias del paquete pip y las dependencias del complemento labextension de JupyterLab, y los números de versión deben ser compatibles.

Los siguientes ejemplos usan 0.27.0. Verifique las versiones disponibles a continuación para usar la última.

Laboratorio Jupyter 1.2.x

pip install tensorflow_model_analysis==0.27.0
jupyter labextension install tensorflow_model_analysis@0.27.0
jupyter labextension install @jupyter-widgets/jupyterlab-manager@1.1

Laboratorio Jupyter 2

pip install tensorflow_model_analysis==0.27.0
jupyter labextension install tensorflow_model_analysis@0.27.0
jupyter labextension install @jupyter-widgets/jupyterlab-manager@2

Solución de problemas

Verifique los paquetes de pip:

pip list

Comprobar extensiones:

jupyter labextension list

Dependencias notables

Se requiere TensorFlow.

Se requiere Apache Beam ; es la forma en que se admite la computación distribuida eficiente. De forma predeterminada, Apache Beam se ejecuta en modo local, pero también se puede ejecutar en modo distribuido mediante Google Cloud Dataflow y otros ejecutores de Apache Beam .

También se requiere Apache Arrow . TFMA usa Arrow para representar datos internamente con el fin de hacer uso de funciones numpy vectorizadas.

Empezando

Para obtener instrucciones sobre el uso de TFMA, consulte la guía de inicio .

Versiones compatibles

La siguiente tabla muestra las versiones del paquete TFMA que son compatibles entre sí. Esto está determinado por nuestro marco de prueba, pero también pueden funcionar otras combinaciones no probadas .

Preguntas

Dirija cualquier pregunta sobre cómo trabajar con TFMA a Stack Overflow mediante la etiqueta tensorflow-model-analysis .