hdallatorre commited on
Commit
64904ca
1 Parent(s): 5ef47a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -96,7 +96,7 @@ probabilities = torch.nn.functional.softmax(logits, dim=-1)
96
  print(f"Probabilities shape: {probabilities.shape}")
97
 
98
  # Get probabilities associated with intron
99
- idx_intron = features.index("intron")
100
  probabilities_intron = probabilities[:,:,idx_intron]
101
  print(f"Intron probabilities shape: {probabilities_intron.shape}")
102
 
 
96
  print(f"Probabilities shape: {probabilities.shape}")
97
 
98
  # Get probabilities associated with intron
99
+ idx_intron = model.config.features.index("intron")
100
  probabilities_intron = probabilities[:,:,idx_intron]
101
  print(f"Intron probabilities shape: {probabilities_intron.shape}")
102