再帰的ニューラルネットワーク(RNN, LSTM)をTensorFlowを用いて実装しました#1

TensorFlowを用いた再帰的ニューラルネットワークのクラスを実装したので晒していきます。数回に分けてクラスの概要、機能、実装の説明をしようと思います。バグや実装の悪い部分などがあればTwitterやコメント欄にメッセージをくださると助かります。

TensorFlowを使った自作ラッパーを作成しました

Abstract

TensorFlowそのものの動作を学ぶために作った機械学習用の自作ラッパーを紹介します。現在はいくつか機能(RNN対応、活性化関数の追加)が増えているのですが、今回のエントリーでは一番初めにベースとなったDNNの自作ラッパーをもとに設計の時に考えたこと文章として残しておきます。

ラッパー作成に関してはTensorFlowやchainerのフロントエンドとして使われているKerasのインターフェイスをある程度参考に実装を行っています。

Scroll to top