Analyse du modèle TensorFlow

TensorFlow Model Analysis (TFMA) est une bibliothèque permettant d'évaluer les modèles TensorFlow. Il permet aux utilisateurs d'évaluer leurs modèles sur de grandes quantités de données de manière distribuée, en utilisant les mêmes métriques définies dans leur entraîneur. Ces métriques peuvent être calculées sur différentes tranches de données et visualisées dans les notebooks Jupyter.

Installation

La méthode recommandée pour installer TFMA consiste à utiliser le package PyPI :

pip install tensorflow-model-analysis

Construire TFMA à partir des sources

Pour construire à partir des sources, suivez les étapes suivantes :

Installez le protocole selon le lien mentionné : protoc

Créez un environnement virtuel en exécutant les commandes

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

Cela construira la roue TFMA dans le répertoire dist. Pour installer la roue à partir du répertoire dist, exécutez les commandes

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

Forfaits de nuit

TFMA héberge également des packages nocturnes sur https://pypi-nightly.tensorflow.org sur Google Cloud. Pour installer le dernier package nocturne, veuillez utiliser la commande suivante :

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

Cela installera les packages nocturnes pour les principales dépendances de TFMA telles que TensorFlow Metadata (TFMD), TFX Basic Shared Libraries (TFX-BSL).

Actuellement, TFMA nécessite que TensorFlow soit installé mais n'a pas de dépendance explicite sur le package TensorFlow PyPI. Consultez les guides d'installation de TensorFlow pour obtenir des instructions.

Pour activer la visualisation TFMA dans Jupyter Notebook :

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

Laboratoire Jupyter

Au moment de la rédaction, à cause de https://github.com/pypa/pip/issues/9187, pip install pourrait ne jamais se terminer. Dans ce cas, vous devez revenir à la version 19 au lieu de 20 : pip install "pip<20" .

L'utilisation d'une extension JupyterLab nécessite l'installation de dépendances sur la ligne de commande. Vous pouvez le faire dans la console de l'interface utilisateur de JupyterLab ou sur la ligne de commande. Cela inclut l'installation séparée de toutes les dépendances du package pip et des dépendances du plugin JupyterLab labextension, et les numéros de version doivent être compatibles.

Les exemples ci-dessous utilisent 0.27.0. Vérifiez les versions disponibles ci-dessous pour utiliser la dernière.

Laboratoire 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

Laboratoire 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

Dépannage

Vérifiez les packages pip :

pip list

Vérifiez les extensions :

jupyter labextension list

Dépendances notables

TensorFlow est requis.

Apache Beam est requis ; c'est ainsi que le calcul distribué efficace est pris en charge. Par défaut, Apache Beam s'exécute en mode local mais peut également s'exécuter en mode distribué à l'aide de Google Cloud Dataflow et d'autres exécuteurs Apache Beam.

Apache Arrow est également requis. TFMA utilise Arrow pour représenter les données en interne afin d'utiliser des fonctions numpy vectorisées.

Commencer

Pour obtenir des instructions sur l'utilisation de TFMA, consultez le guide de démarrage .

Versions compatibles

Le tableau suivant présente les versions du package TFMA compatibles les unes avec les autres. Ceci est déterminé par notre cadre de test, mais d'autres combinaisons non testées peuvent également fonctionner.

Questions

Veuillez adresser toutes vos questions sur l'utilisation de TFMA à Stack Overflow à l'aide de la balise tensorflow-model-analysis .