Kerasの正規化または脱落層で訓練を受けたときの予測方法は？

Question

正規化レイヤーでモデルを訓練しました。トレーニングフェーズで

：推論時間

model=Sequential() 
model.add() 

... 

k.set_learning_phase(1) 
ModelCheckpoint(weights_file) 
model.fit() 

：

k.set_learning_phase(0) 
model.load_weights(weights_file) 
model.predict_classes() 

... 

Kerasのバージョン：2.0.8のコードは、このようになります。 CaffeでSegNetを使用するようなトレーニングをした後、BNを計算するための特別なコードが必要ですか？

Answer 1

いいえ、バッチ正規化レイヤーまたはドロップアウトレイヤーを使用する場合は、特別な操作を行う必要はありません。 Kerasはすでに学習/テストの段階を追跡していますので、predictまたはpredict_classesを使用すると、正しいことが実行されます。

学習フェーズを手動で設定する必要はありません。Kerasは既にそれを実行しています。