From ef90137c4c4251cccffc27811fc2f341a0116e40 Mon Sep 17 00:00:00 2001 From: Neil Lawrence Date: Wed, 19 May 2021 10:06:40 +0100 Subject: [PATCH] Fix dimensionality reduction example to work with notebook. --- GPy/examples/dimensionality_reduction.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/GPy/examples/dimensionality_reduction.py b/GPy/examples/dimensionality_reduction.py index 634f9790..9218a0e8 100644 --- a/GPy/examples/dimensionality_reduction.py +++ b/GPy/examples/dimensionality_reduction.py @@ -685,17 +685,23 @@ def cmu_mocap(subject='35', motion=['01'], in_place=True, optimize=True, verbose # Make figure move in place. data['Y'][:, 0:3] = 0.0 Y = data['Y'] - Y_mean = Y.mean(0) - Y_std = Y.std(0) - m = GPy.models.GPLVM((Y - Y_mean) / Y_std, 2) + m = GPy.models.GPLVM(Y, 2, normalizer=True) if optimize: m.optimize(messages=verbose, max_f_eval=10000) if plot: - fig, (latent_axes, sense_axes) = plt.subplots(1, 2) + fig, _ = plt.subplots(figsize=(8, 5)) + latent_axes = fig.add_subplot(131) + sense_axes = fig.add_subplot(132) + viz_axes = fig.add_subplot(133, projection='3d') + + m.plot_latent(ax=latent_axes) + latent_axes.set_aspect('equal') + y = m.Y[0, :] - data_show = GPy.plotting.matplot_dep.visualize.skeleton_show(y[None, :], data['skel']) - lvm_visualizer = GPy.plotting.matplot_dep.visualize.lvm(m.X[0].copy(), m, data_show, latent_axes=latent_axes) + data_show = GPy.plotting.matplot_dep.visualize.skeleton_show(y[None, :], data['skel'], viz_axes) + + lvm_visualizer = GPy.plotting.matplot_dep.visualize.lvm(m.X[0].copy(), m, data_show, latent_axes=latent_axes, sense_axes=sense_axes) input('Press enter to finish') lvm_visualizer.close() data_show.close()