4065c4b2cc1ebe2eeb829e8ff3f620a4f510b096,nn_wtf/neural_network_graph.py,NeuralNetworkGraph,predict,#NeuralNetworkGraph#Any#Any#,123

Before Change

            self.prediction_op = tf.argmax(self.layers[-1], 1)

        image_data = image.reshape(self.input_size)
        feed_dict = {self.input_placeholder: [image_data]}
        best = session.run(self.prediction_op, feed_dict)
        return best[0]


    def _add_layer(self, layer_name, in_units_size, out_units_size, input_layer, function=lambda x: x):
        with tf.name_scope(layer_name):

After Change

    def predict(self, session, image):
        if self.predictor is None:
            self.predictor = Predictor(self)
        return self.predictor.predict(session, image)

    def _add_layer(self, layer_name, in_units_size, out_units_size, input_layer, function=lambda x: x):
        with tf.name_scope(layer_name):
            weights = self._initialize_weights(in_units_size, out_units_size)

Instances