classifier_tf.py 文件源码

def predict_proba_with_loss(self, X, y):
        feed_dict = {}
        feed_dict[self.labels] = y
        for key, tensor in self.features.items():
            feed_dict[tensor] = X[key]
        prediction, loss = tf.get_default_session().run(
            [self.prediction, self.loss], feed_dict=feed_dict)
        return np.reshape(prediction, [-1]), loss