diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-04 00:08:07 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-04 00:08:07 -0700 |
| commit | 9d33a74915382159ce97e3b4142743a7e3e1c72d (patch) | |
| tree | 82fcf302fc9458f25568db9fdcc794cc9839b615 | |
| parent | 6d2a0a621f00af03ec276cf72ab258c4f482e8cb (diff) | |
Fix tf issue
| -rw-r--r-- | ml_exp/do_ml.py | 4 |
1 files 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)) |