fix nonparametric

better · Apr 6, 2018 · 608816b · 608816b
1 parent 18505fd
commit 608816b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/convoys/single.py b/convoys/single.py
@@ -115,9 +115,9 @@ def get_LL(log_survived_until, log_survived_after, log_observed):
             # elements are almost zero.
             self.params = {
                 'beta': sess.run(beta),
-                'beta_std': tf_utils.get_hessian(sess, LL, {}, beta) ** -0.5,
+                'beta_std': sess.run(tf.hessians(-LL, [beta])[0]) ** -0.5,
                 'z': sess.run(z),
-                'z_std': numpy.diag(tf_utils.get_hessian(sess, LL, {}, z)) ** -0.5,  # TODO: seems inefficient
+                'z_std': numpy.diag(sess.run(tf.hessians(-LL, [z])[0])) ** -0.5,  # TODO: seems inefficient
             }
             self.params['z_std'] = 0  # not sure what's up, will revisit this