jheuschkel commited on
Commit
d40173d
·
verified ·
1 Parent(s): e6ff2b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -50,7 +50,7 @@ model = CodonEmbeddings() #this loads the model & tokenizer using our built-in f
50
 
51
  seq = 'ATGTCCACCGGGCGGTGA'
52
 
53
- mean_pooled_embedding = model.get_mean_embedding(seq, species_token_type=67) #E. coli
54
  #returns --> tensor of shape [768]
55
 
56
  raw_output = model.get_raw_embeddings(seq, species_token_type=67) #E. coli
 
50
 
51
  seq = 'ATGTCCACCGGGCGGTGA'
52
 
53
+ mean_pooled_embedding = model.get_mean_embedding(seq, species_token_type=30) #E. coli
54
  #returns --> tensor of shape [768]
55
 
56
  raw_output = model.get_raw_embeddings(seq, species_token_type=67) #E. coli