TFLearn может быть определен как модульный и прозрачный аспект глубокого обучения, используемый в платформе TensorFlow. Основным мотивом TFLearn является предоставление API TensorFlow более высокого уровня для облегчения и демонстрации новых экспериментов.
Рассмотрим следующие важные особенности TFLearn:
Установите TFLearn, выполнив следующую команду:
pip install tflearn
После выполнения кода выше будет сгенерирован следующий вывод:
На следующем рисунке показана реализация TFLearn с классификатором Random Forest:
from __future__ import division, print_function, absolute_import # реализация модуля TFLearn import tflearn from tflearn.estimators import RandomForestClassifier # Загрузка и предварительная обработка данных в отношении набора данных import tflearn.datasets.mnist as mnist X, Y, testX, testY = mnist.load_data(one_hot = False) m = RandomForestClassifier(n_estimators = 100, max_nodes = 1000) m.fit(X, Y, batch_size = 10000, display_step = 10) print("Вычислить точность данных о поездах:") print(m.evaluate(X, Y, tflearn.accuracy_op)) print("Вычислить точность на тестовом наборе данных:") print(m.evaluate(testX, testY, tflearn.accuracy_op)) print("Цифры для тестовых изображений id от 0 до 5:") print(m.predict(testX[:5])) print("Истинные цифры:") print(testY[:5])