[ard plotting] adjustments to the filtering

This commit is contained in:
mzwiessele 2014-09-05 14:17:00 +01:00
parent 470c0dcfe6
commit 3fff0448be
2 changed files with 4 additions and 4 deletions

View file

@ -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
#---------------------------------------#

View file

@ -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,:]