AmelieSchreiber
commited on
Commit
•
4bb0bd0
1
Parent(s):
5b925e6
Update README.md
Browse files
README.md
CHANGED
@@ -12,9 +12,15 @@ tags:
|
|
12 |
|
13 |
# ESM-2 (`esm2_t6_8M_UR50D`)
|
14 |
|
15 |
-
This is a fine-tuned version of ESM-2 for sequence classification
|
16 |
-
two classes, either "cystolic" or "membrane".
|
17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
To use try running:
|
19 |
```python
|
20 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
|
12 |
|
13 |
# ESM-2 (`esm2_t6_8M_UR50D`)
|
14 |
|
15 |
+
This is a fine-tuned version of [ESM-2](https://huggingface.co/facebook/esm2_t6_8M_UR50D) for sequence classification
|
16 |
+
that categorizes protein sequences into two classes, either "cystolic" or "membrane".
|
17 |
|
18 |
+
## Training and Accuracy
|
19 |
+
|
20 |
+
The model is trained using [this notebook](https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/protein_language_modeling.ipynb)
|
21 |
+
and achieved an eval accuracy of 94.83163664839468 %.
|
22 |
+
|
23 |
+
## Using the Model
|
24 |
To use try running:
|
25 |
```python
|
26 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|