Update README.md
Browse files
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=
|
| 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
|