TFLearn:為TensorFlow提供更高級別的API 的深度學習庫

  • 2019 年 10 月 4 日
  • 筆記

譯者 | fendouai

【磐創AI 導讀】:本文將會帶大家了解TFLearn,歡迎大家轉發、留言。

TFlearn是一個基於Tensorflow構建的模組化透明深度學習庫。它旨在為TensorFlow提供更高級別的API,以促進和加速實驗,同時保持完全透明並與之兼容。

TFLearn功能包括:

  • 通過教程和示例,易於使用和理解用於實現深度神經網路的高級API。
  • 通過高度模組化的內置神經網路層,正則化器,優化器,指標進行快速原型設計
  • Tensorflow完全透明。所有功能都是通過張量構建的,可以獨立於TFLearn使用。
  • 強大的輔助功能,可以訓練任何TensorFlow 圖,支援多個輸入,輸出和優化器。
  • 簡單而美觀的圖形可視化,包含有關權重,梯度,激活等的詳細資訊。
  • 輕鬆使用多個CPU / GPU的設備。

高級API目前支援大多數最近的深度學習模型,如Convolutions,LSTM,BiRNN,BatchNorm,PReLU,殘留網路,生成網路……未來,TFLearn也將與最新版本保持同步最新的深度學習模型。

注意:最新的TFLearn(v0.3)僅與TensorFlow v1.0及更高版本兼容。

概覽

#分類

# Classification  tflearn.init_graph(num_cores=8, gpu_memory_fraction=0.5)    net = tflearn.input_data(shape=[None, 784])  net = tflearn.fully_connected(net, 64)  net = tflearn.dropout(net, 0.5)  net = tflearn.fully_connected(net, 10, activation='softmax')  net = tflearn.regression(net, optimizer='adam', loss='categorical_crossentropy')    model = tflearn.DNN(net)  model.fit(X, Y)

#序列生成

net = tflearn.input_data(shape=[None, 100, 5000])  net = tflearn.lstm(net, 64)  net = tflearn.dropout(net, 0.5)  net = tflearn.fully_connected(net, 5000, activation='softmax')  net = tflearn.regression(net, optimizer='adam', loss='categorical_crossentropy')    model = tflearn.SequenceGenerator(net, dictionary=idx, seq_maxlen=100)  model.fit(X, Y)  model.generate(50, temperature=1.0)

這裡有更多的例子 http://tflearn.org/examples/。

TensorFlow安裝

TFLearn需要安裝Tensorflow(版本1.0+)。要安裝TensorFlow,只需運行:

pip install tensorflow

或者,支援GPU:

pip install tensorflow-gpu

有關更多詳細資訊,請參閱TensorFlow安裝說明。

TFLearn安裝

要安裝TFLearn,最簡單的方法就是運行。對於前沿版本(推薦):

pip install git + https://github.com/tflearn/tflearn.git

對於最新的穩定版本:

pip install tflearn

否則,您也可以通過運行(從源文件夾)從源安裝:

python setup.py install

有關詳細資訊,請參閱「安裝指南」。

入門

請參閱TFLearn入門,了解TFLearn基本功能或開始瀏覽TFLearn教程。 http://tflearn.org/getting_started/

例子

有許多可用的神經網路實現,請參見示例。 http://tflearn.org/examples/

文檔

http://tflearn.org/doc_index

模型可視化

損失可視化

圖層可視化

【1】tflearn:https://github.com/tflearn/tflearn