mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-15 06:52:39 +02:00
[kernel ard plot] label adjustment
This commit is contained in:
parent
8cf11257b0
commit
7bdbe21cd1
1 changed files with 5 additions and 2 deletions
|
|
@ -68,6 +68,8 @@ def plot_ARD(kernel, fignum=None, ax=None, title='', legend=False, filtering=Non
|
||||||
|
|
||||||
ard_params = np.atleast_2d(kernel.input_sensitivity(summarize=False))
|
ard_params = np.atleast_2d(kernel.input_sensitivity(summarize=False))
|
||||||
bottom = 0
|
bottom = 0
|
||||||
|
last_bottom = bottom
|
||||||
|
|
||||||
x = np.arange(kernel.input_dim)
|
x = np.arange(kernel.input_dim)
|
||||||
|
|
||||||
if order is None:
|
if order is None:
|
||||||
|
|
@ -77,12 +79,13 @@ def plot_ARD(kernel, fignum=None, ax=None, title='', legend=False, filtering=Non
|
||||||
if kernel.parameters[i].name in order:
|
if kernel.parameters[i].name in order:
|
||||||
c = Tango.nextMedium()
|
c = Tango.nextMedium()
|
||||||
bars.append(plot_bars(fig, ax, x, ard_params[i,:], c, kernel.parameters[i].name, bottom=bottom))
|
bars.append(plot_bars(fig, ax, x, ard_params[i,:], c, kernel.parameters[i].name, bottom=bottom))
|
||||||
bottom += ard_params[i,:]
|
last_bottom = ard_params[i,:]
|
||||||
|
bottom += last_bottom
|
||||||
else:
|
else:
|
||||||
print "filtering out {}".format(kernel.parameters[i].name)
|
print "filtering out {}".format(kernel.parameters[i].name)
|
||||||
|
|
||||||
ax.set_xlim(-.5, kernel.input_dim - .5)
|
ax.set_xlim(-.5, kernel.input_dim - .5)
|
||||||
add_bar_labels(fig, ax, [bars[-1]], bottom=bottom-ard_params[i,:])
|
add_bar_labels(fig, ax, [bars[-1]], bottom=bottom-last_bottom)
|
||||||
|
|
||||||
if legend:
|
if legend:
|
||||||
if title is '':
|
if title is '':
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue