Update README.md
Browse files
README.md
CHANGED
|
@@ -38,10 +38,6 @@ logits = torch.squeeze(gsfm(token_ids))
|
|
| 38 |
top_10 = sorted(zip(logits, vocab.vocab))[-10:]
|
| 39 |
top_10
|
| 40 |
|
| 41 |
-
# get gene embedding
|
| 42 |
-
gene_embeddings = gsfm.embedding(token_ids)
|
| 43 |
-
gene_embeddings
|
| 44 |
-
|
| 45 |
# get model middle layer
|
| 46 |
gene_set_encoding = gsfm.encode(token_ids)
|
| 47 |
gene_set_encoding
|
|
|
|
| 38 |
top_10 = sorted(zip(logits, vocab.vocab))[-10:]
|
| 39 |
top_10
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
# get model middle layer
|
| 42 |
gene_set_encoding = gsfm.encode(token_ids)
|
| 43 |
gene_set_encoding
|