softmax conversion of logits in TfModelBase.predict_proba

cgpotts · May 14, 2018 · 470ac06 · 470ac06
1 parent 9d07960
commit 470ac06
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tf_model_base.py b/tf_model_base.py
@@ -171,8 +171,9 @@ def predict_proba(self, X):
         of X and n is the number of classes
 
         """
+        self.probs = tf.nn.softmax(self.model)
         return self.sess.run(
-            self.model, feed_dict=self.test_dict(X))
+            self.probs, feed_dict=self.test_dict(X))
 
     def predict(self, X):
         """Return classifier predictions, as the class with the