tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning
Apache License 2.0
3.18k stars 439 forks source link

Turn off dropout at predict time #79

Closed ilblackdragon closed 8 years ago

ilblackdragon commented 8 years ago

Currently, dropout is still applied at test time, which leads to incorrect results.

The proposed solution is to gather all dropout probability nodes and feed dict the probability 0 when run predict.