TFLearn:特色的深度學習函式庫,TensorFlow 高階 API。

TFlearn 是建構在 TensorFlow 的模組化且透明化的深度學習函式庫。專門設計用來提升 TensorFlow 的高階 API,以便簡化並加速實驗,同時保持完全透明化且與之相容。

TFLearn 的特色包含

  • 易於使用且能理解的高階 API 來實作深度神經網路,並提供教學範例。
  • 透過高度模組化的內建神經網路圖層、正規器、最佳化器、指標,來快速建立雛形。
  • TensorFlow 的完全透明化。所有元件均建構自張量,並可獨立於 TFLearn 使用。
  • 訓練任何 TensorFlow 圖形的強力幫手函式,並支援多重輸入、輸出與最佳化器。
  • 容易且美觀的圖形呈現,包含有關於權重、梯度、啟用等詳細資訊...
  • 輕而易舉的裝置配置,適用於多重 CPU/GPU。

高階的 API 目前支援絕大多數近期深度學習模型,例如:卷積、LSTM、雙向迴歸神經網路、批次正規化、ReLU、殘餘網路、生成網路等...。而 TFLearn 未來也預計會持續更新至最新的深度學習技術。

注意:最新的 TFLearn(v0.3)只與 TensorFlow 1.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)
# Sequence Generation
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)

還有更多範例在此處 提供

從何開始?

若要安裝 TFLearn,請參閱:安裝指南

若您的 TensorFlow 版本低於 1.0:升級 TensorFlow

開始的步驟:開始使用 TFLearn 程序TFLearn 教學

更多範例:範例清單

欲瀏覽 API,請參考 API 文件

模型視覺化

圖形

Graph Visualization

損失與精準度(多次回合執行)

Loss Visualization

圖層

Layers Visualization

來源

GitHub:https://github.com/tflearn/tflearn

貢獻

這是 TFLearn 的第一個版本,如果您發現任何錯誤,請在 GitHub 問題部分回報。

非常歡迎改善和新功能的要求!請不要猶豫,扭曲並調整 TF Learn,並傳送拉取要求。

更多資訊:貢獻給 TFLearn

授權

MIT 授權