SdA_wm.py 文件源码

def predict(self, test_set_x):
        predict_model = theano.function(inputs=[self.x], outputs=self.pred)
        predict_proba = predict_model(test_set_x)
        predict = numpy.argmax(predict_proba, axis=1)
        return predict