Visualisation of Machine Learning Training and Interpretation

Yellowbrick

Утилита для визуализации данных в машинном обучении. Обертка над sklearn, позволяющая выводить сразу метрики качества, автоматически выполнять отбор признаков, для моделей из sklearn.

• Документация.
• Пример в ноутбуке.

Local Interpretable Model-agnostic Explanations (LIME)

Метод предназначен для интерпретации моделей машинного обучения. Работает на основе ресемплинга данных в окрестности некоторой точки и постронения локально линейно моделей. На основе построенной линейной модели получается важность того или иного признака объекта. Для текста это важность того или иного слова. Для изображения это важность той или иной части изображения.

• Статья.
• Документация.
• Пример в ноутбуке.

TensorBoard

Утилита для отслеживания моделей во время их обучения (оптимизации параметров).

• Документация.
• Пример для LSTM.
• Пример для CNN.
• Пример для FNN.