Martijn van Beers commited on
Commit
9c33733
1 Parent(s): 97c0e0a

Fix copied visualize_text

Browse files

format_classname is from captum.visualization too, so since we call it
outside captum, we need to add the namespace to the call

Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -104,14 +104,14 @@ def visualize_text(datarecords, legend=True):
104
  "".join(
105
  [
106
  "<tr>",
107
- format_classname(datarecord.true_class),
108
- format_classname(
109
  "{0} ({1:.2f})".format(
110
  datarecord.pred_class, datarecord.pred_prob
111
  )
112
  ),
113
- format_classname(datarecord.attr_class),
114
- format_classname("{0:.2f}".format(datarecord.attr_score)),
115
  format_word_importances(
116
  datarecord.raw_input_ids, datarecord.word_attributions
117
  ),
 
104
  "".join(
105
  [
106
  "<tr>",
107
+ visualization.format_classname(datarecord.true_class),
108
+ visualization.format_classname(
109
  "{0} ({1:.2f})".format(
110
  datarecord.pred_class, datarecord.pred_prob
111
  )
112
  ),
113
+ visualization.format_classname(datarecord.attr_class),
114
+ visualization.format_classname("{0:.2f}".format(datarecord.attr_score)),
115
  format_word_importances(
116
  datarecord.raw_input_ids, datarecord.word_attributions
117
  ),