diff --git a/GPy/models_modules/bayesian_gplvm.py b/GPy/models_modules/bayesian_gplvm.py index 74eeb4b7..1ac6ba1e 100644 --- a/GPy/models_modules/bayesian_gplvm.py +++ b/GPy/models_modules/bayesian_gplvm.py @@ -215,12 +215,13 @@ class BayesianGPLVM(SparseGP, GPLVM): from matplotlib.cm import get_cmap from GPy.util.latent_space_visualizations.controllers.imshow_controller import ImAnnotateController + if not 'cmap' in kwargs.keys(): + kwargs.update(cmap=get_cmap('jet'),)) controller = ImAnnotateController(ax, plot_function, tuple(self.X.min(0)[:, significant_dims]) + tuple(self.X.max(0)[:, significant_dims]), resolution=resolution, aspect=aspect, - cmap=get_cmap('jet'), **kwargs) ax.legend() ax.figure.tight_layout()