Skip to content

Commit c5a43f5

Browse files
authored
Update trainer index references in README
1 parent 13428b2 commit c5a43f5

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -180,11 +180,11 @@ motif_cav_trainers = torch.load("motif_cav_trainers.pt")
180180
bed_cav_trainer = torch.load("bed_cav_trainer.pt")
181181

182182
# inspect trainer properties
183-
cav_trainer = motif_cav_trainers[0] # here we take the first motif cav trainer that correponds to the first number of motif insertions
183+
cav_trainer = motif_cav_trainers[4] # here we take the first motif cav trainer that correponds to the first number of motif insertions (4 insertions)
184184
# retrieve F-scores
185-
motif_cav_trainers[0].cav_fscores
185+
motif_cav_trainers[4].cav_fscores
186186
# retrieve CAVs
187-
motif_cav_trainers[0].cav_weights
187+
motif_cav_trainers[4].cav_weights
188188
```
189189

190190
You can also retrieve the model decorated by TPCAV parameters by

0 commit comments

Comments
 (0)