From 3fff0448be46910fc7324bd58900639473fd5bcf Mon Sep 17 00:00:00 2001 From: mzwiessele Date: Fri, 5 Sep 2014 14:17:00 +0100 Subject: [PATCH] [ard plotting] adjustments to the filtering --- GPy/kern/_src/linear.py | 2 +- GPy/plotting/matplot_dep/kernel_plots.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/GPy/kern/_src/linear.py b/GPy/kern/_src/linear.py index 9fdacdbb..c30e344e 100644 --- a/GPy/kern/_src/linear.py +++ b/GPy/kern/_src/linear.py @@ -103,7 +103,7 @@ class Linear(Kern): def gradients_X_diag(self, dL_dKdiag, X): return 2.*self.variances*dL_dKdiag[:,None]*X - def input_sensitivity(self): + def input_sensitivity(self, summarize=True): return np.ones(self.input_dim) * self.variances #---------------------------------------# diff --git a/GPy/plotting/matplot_dep/kernel_plots.py b/GPy/plotting/matplot_dep/kernel_plots.py index db43f147..f2082db0 100644 --- a/GPy/plotting/matplot_dep/kernel_plots.py +++ b/GPy/plotting/matplot_dep/kernel_plots.py @@ -72,11 +72,11 @@ def plot_ARD(kernel, fignum=None, ax=None, title='', legend=False, filtering=Non x = np.arange(kernel.input_dim) - if order is None: - order = kernel.parameter_names(recursive=False) + if filtering is None: + filtering = kernel.parameter_names(recursive=False) for i in range(ard_params.shape[0]): - if kernel.parameters[i].name in order: + if kernel.parameters[i].name in filtering: c = Tango.nextMedium() bars.append(plot_bars(fig, ax, x, ard_params[i,:], c, kernel.parameters[i].name, bottom=bottom)) last_bottom = ard_params[i,:]