K.function的用法

烈酒焚心 提交于 2020-03-03 11:58:01

K.function除了能够实现指定输入和输出之外,还能够决定网络是处于训练模式还是测试模型

1或True:训练模式,dropout开启
0或False:测试模式,dropout失活

代码如下,需要先用K.function“实例化”一个f函数,然后f((X_test, True))代表将X_test传入作为model.layers[0]的输入,True代表训练模式——dropout打开,最终是model.output作为输出

    f = K.function([self.model.layers[0].input, 
                    K.symbolic_learning_phase()], 
                    [self.model.output])

    mc_preds = np.array([f((X_test, True))[0] for _ in range(mc_samples)])
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!