【Keras】Kerasで中間層を分割するモデルの作成

今回はKerasで中間の層を分割するモデルの作成方法についてお話します。

目標としては下の図のようにすることです。

f:id:ttt242242:20190718052519j:plain

元のモデル

まずは元のモデルを作成します。

f:id:ttt242242:20190718052539p:plain

中間層を分割したモデル

Keras のLambdaを用いて構成します。

上記のコードによって、以下のようなモデルになります。

f:id:ttt242242:20190718052553p:plain

7行目のLambdaについて少し説明します。

上記のLambdaではinputs層のoutputを引数xとしています。
そして、lambdaの中でxの前半部分をx[:, :5]を入力として受け取っています。
ちなみに、x[:, :5]の前半の[:] はバッチ用ですので、必ずつけなければなりません。

参考文献

Guide to the Functional API - Keras Documentation
kerasで頭に描いたネットワーク構造を実現するためのTips ~ Lambda 編 ~ - Qiita
# 追記2019/10/01 消費税が10%に上がりました。 ではなく、Tensorflow2.0が正式リリースされました。少し前からKerasはTensorflowにインクルードされていますが、そのKerasにおいて ``` R...

コメント

タイトルとURLをコピーしました