如何使用 Keras 和 Python 從模型中移除一個層？

Tensorflow 是 Google 提供的一個機器學習框架。它是一個開源框架，與 Python 結合使用以實現演算法、深度學習應用程式等等。它用於研究和生產目的。

Keras 是作為 ONEIROS 專案（開放式神經電子智慧機器人作業系統）研究的一部分開發的。Keras 是一個深度學習 API，是用 Python 編寫的。它是一個高階 API，具有高效的介面，有助於解決機器學習問題。

它具有高度的可擴充套件性，並具有跨平臺功能。這意味著 Keras 可以執行在 TPU 或 GPU 叢集上。Keras 模型還可以匯出以在 Web 瀏覽器或手機上執行。

Keras 已經存在於 Tensorflow 包中。可以使用以下程式碼行訪問它。

import tensorflow
from tensorflow import keras

我們正在使用 Google Colaboratory 來執行以下程式碼。Google Colab 或 Colaboratory 幫助在瀏覽器上執行 Python 程式碼，無需任何配置即可免費訪問 GPU（圖形處理單元）。Colaboratory 是建立在 Jupyter Notebook 之上的。

以下是移除層的程式碼：

示例

print("Removing layers using the pop function")
model.pop()
print("The current number of layers in the model after eliminating one layer")
print(len(model.layers))

程式碼來源： https://www.tensorflow.org/guide/keras/sequential_model

輸出

Removing layers using the pop function
The current number of layers in the model after eliminating one layer
2

解釋

  • 可以透過使用點運算子將模型名稱與函式關聯來呼叫“pop”函式。

  • 完成此操作後，可以檢查層的長度。

  • 這將有助於確認確實刪除了一層。

更新於： 2021年1月21日

2K+ 閱讀量

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告