aksell commited on
Commit
01ba5a1
·
1 Parent(s): 528cd6e

Show chain in table of top residues

Browse files
hexviz/🧬Attention_Visualization.py CHANGED
@@ -287,12 +287,12 @@ for att_weight, _, chain, resi in top_residues:
287
  res = chain_dict[chain][resi]
288
  except KeyError:
289
  continue
290
- el = (att_weight, f"{res.resname:3}{res.id[1]}")
291
  data.append(el)
292
 
293
  df = pd.DataFrame(data, columns=["Total attention (disregarding direction)", "Residue"])
294
  st.markdown(
295
- f"The {n_highest_resis} residues with the highest attention sums are labeled in the visualization and listed here:"
296
  )
297
  st.table(df)
298
 
 
287
  res = chain_dict[chain][resi]
288
  except KeyError:
289
  continue
290
+ el = (att_weight, f"{res.resname:3}{res.id[1]}({chain})")
291
  data.append(el)
292
 
293
  df = pd.DataFrame(data, columns=["Total attention (disregarding direction)", "Residue"])
294
  st.markdown(
295
+ f"The {n_highest_resis} residues (per chain) with the highest attention sums are labeled in the visualization and listed here:"
296
  )
297
  st.table(df)
298