Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def visualize(e, plot_cmap="RdBu", matplotlib=False, figsize=(20,3), show=True, ordering_keys=None, ordering_keys_time_format=None, text_rotation=0):
plot_cmap = verify_valid_cmap(plot_cmap)
if isinstance(e, AdditiveExplanation):
if matplotlib:
return AdditiveForceVisualizer(e, plot_cmap=plot_cmap).matplotlib(figsize=figsize, show=show, text_rotation=text_rotation)
else:
return AdditiveForceVisualizer(e, plot_cmap=plot_cmap).html()
elif isinstance(e, Explanation):
if matplotlib:
assert False, "Matplotlib plot is only supported for additive explanations"
else:
return SimpleListVisualizer(e).html()
elif isinstance(e, Sequence) and len(e) > 0 and isinstance(e[0], AdditiveExplanation):
if matplotlib:
assert False, "Matplotlib plot is only supported for additive explanations"
else:
return AdditiveForceArrayVisualizer(e, plot_cmap=plot_cmap, ordering_keys=ordering_keys, ordering_keys_time_format=ordering_keys_time_format).html()
else:
assert False, "visualize() can only display Explanation objects (or arrays of them)!"
def visualize(e, plot_cmap="RdBu", matplotlib=False, figsize=(20,3), show=True, ordering_keys=None, ordering_keys_time_format=None, text_rotation=0):
plot_cmap = verify_valid_cmap(plot_cmap)
if isinstance(e, AdditiveExplanation):
if matplotlib:
return AdditiveForceVisualizer(e, plot_cmap=plot_cmap).matplotlib(figsize=figsize, show=show, text_rotation=text_rotation)
else:
return AdditiveForceVisualizer(e, plot_cmap=plot_cmap).html()
elif isinstance(e, Explanation):
if matplotlib:
assert False, "Matplotlib plot is only supported for additive explanations"
else:
return SimpleListVisualizer(e).html()
elif isinstance(e, Sequence) and len(e) > 0 and isinstance(e[0], AdditiveExplanation):
if matplotlib:
assert False, "Matplotlib plot is only supported for additive explanations"
else:
return AdditiveForceArrayVisualizer(e, plot_cmap=plot_cmap, ordering_keys=ordering_keys, ordering_keys_time_format=ordering_keys_time_format).html()
else:
assert False, "visualize() can only display Explanation objects (or arrays of them)!"