Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ pip install --upgrade helical
|
|
37 |
|
38 |
#### Generate Embeddings
|
39 |
```python
|
40 |
-
from helical import HelixmRNA, HelixmRNAConfig
|
41 |
import torch
|
42 |
|
43 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
@@ -57,7 +57,7 @@ embeddings = helix_mrna.get_embeddings(processed_input_data)
|
|
57 |
#### Fine-Tuning
|
58 |
Classification fine-tuning example:
|
59 |
```python
|
60 |
-
from helical import
|
61 |
import torch
|
62 |
|
63 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
37 |
|
38 |
#### Generate Embeddings
|
39 |
```python
|
40 |
+
from helical.models.helix_mrna import HelixmRNA, HelixmRNAConfig
|
41 |
import torch
|
42 |
|
43 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
57 |
#### Fine-Tuning
|
58 |
Classification fine-tuning example:
|
59 |
```python
|
60 |
+
from helical.models.helix_mrna import HelixmRNA, HelixmRNAConfig
|
61 |
import torch
|
62 |
|
63 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|