aksell commited on
Commit
18e02d3
·
1 Parent(s): 51f3451

Tweak interesting heads view

Browse files
hexviz/pages/Identify_interesting_Heads.py CHANGED
@@ -27,12 +27,12 @@ chains = list(structure.get_chains())
27
 
28
  sequence = get_sequence(chains[0])
29
  l = len(sequence)
30
- slice_start= st.sidebar.number_input(f"Slice start(1-{l})",value=1, min_value=1, max_value=l)
31
- slice_end = st.sidebar.number_input(f"Slice end(1-{l})",value=50, min_value=1, max_value=l)
32
  truncated_sequence = sequence[slice_start-1:slice_end]
33
 
34
 
35
- st.text(f"Each tile is a heatmap of attention for the slice {slice_start}:{slice_end} of {pdb_id} (chain A) for one attention head.")
36
 
37
  # TODO: Decide if you should get attention for the full sequence or just the truncated sequence
38
  # Attention values will change depending on what we do.
 
27
 
28
  sequence = get_sequence(chains[0])
29
  l = len(sequence)
30
+ slice_start= st.sidebar.number_input(f"Section start(1-{l})",value=1, min_value=1, max_value=l)
31
+ slice_end = st.sidebar.number_input(f"Section end(1-{l})",value=50, min_value=1, max_value=l)
32
  truncated_sequence = sequence[slice_start-1:slice_end]
33
 
34
 
35
+ st.markdown(f"Each tile is a heatmap of attention for a section of {pdb_id}(A) from residue {slice_start} to {slice_end}. Adjust the section length and starting point in the sidebar.")
36
 
37
  # TODO: Decide if you should get attention for the full sequence or just the truncated sequence
38
  # Attention values will change depending on what we do.