天天看點

Py之textgenrnn:textgenrnn庫的簡介、安裝、使用方法詳細攻略

textgenrnn庫的簡介

textgenrnn 是一個基于 Keras/TensorFlow 的 Python 3 子產品,用于建立 char-rnn,具有許多很酷炫的特性:

它是一個使用注意力權重(attention-weighting)和跳躍嵌入(skip-embedding)等先進技術的現代神經網絡架構,用于加速訓練并提升模型品質。

能夠在字元層級和詞層級上進行訓練和預測。

能夠設定 RNN 的大小、層數,以及是否使用雙向 RNN。

能夠對任何通用的輸入文本檔案進行訓練。

能夠在 GPU 上訓練模型,然後在 CPU 上使用這些模型。

在 GPU 上訓練時能夠使用強大的 CuDNN 實作 RNN,這比标準的 LSTM 實作大大加速了訓練時間。

能夠使用語境标簽訓練模型,能夠更快地學習并在某些情況下産生更好的結果。

textgenrnn庫的安裝

pip install textgenrnn

Py之textgenrnn:textgenrnn庫的簡介、安裝、使用方法詳細攻略

textgenrnn庫的使用方法

後期更新……