From 9d33a74915382159ce97e3b4142743a7e3e1c72d Mon Sep 17 00:00:00 2001 From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com> Date: Wed, 4 Mar 2020 00:08:07 -0700 Subject: Fix tf issue --- ml_exp/do_ml.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ml_exp/do_ml.py b/ml_exp/do_ml.py index ffe9a2d85..5ff3c1105 100644 --- a/ml_exp/do_ml.py +++ b/ml_exp/do_ml.py @@ -93,7 +93,7 @@ def simple_ml(descriptors, shape=(training_size), dtype=tf.float64)) K_tr += dv - # Y_tr = tf.expand_dims(Y_tr, 1) + Y_tr = tf.expand_dims(Y_tr, 1) alpha = tf.linalg.cholesky_solve(tf.linalg.cholesky(K_tr), Y_tr) @@ -104,7 +104,7 @@ def simple_ml(descriptors, sigma, use_tf=use_tf) - # Y_te = tf.expand_dims(Y_te, 1) + Y_te = tf.expand_dims(Y_te, 1) Y_pr = tf.tensordot(K_te, alpha, 1) mae = tf.reduce_mean(tf.abs(Y_pr - Y_te)) -- cgit v1.2.3-70-g09d2