[plotting] magnification plot added

This commit is contained in:
mzwiessele 2015-10-05 14:20:33 +01:00
parent 0610903018
commit 8e80aad902
4 changed files with 67 additions and 4 deletions

View file

@ -357,7 +357,7 @@ def plot_steepest_gradient_map(model, fignum=None, ax=None, which_indices=None,
def plot_function(x):
X[:, significant_dims] = x
dmu_dX = model.dmu_dXnew(X)
dmu_dX = model.predictive_gradients(X)[0]
argmax = np.argmax(dmu_dX, 1)
return dmu_dX[indices, argmax], np.array(labels)[argmax]