Update README.md
Browse files
README.md
CHANGED
|
@@ -26,15 +26,12 @@ https://github.com/andrewdalpino/esm2-function-classifier
|
|
| 26 |
|
| 27 |
## Model Specs
|
| 28 |
|
| 29 |
-
**Vocabulary Size**: 33
|
| 30 |
-
|
| 31 |
-
**
|
| 32 |
-
|
| 33 |
-
**
|
| 34 |
-
|
| 35 |
-
**Encoder Layers**: 12
|
| 36 |
-
|
| 37 |
-
**Context Length**: 1026
|
| 38 |
|
| 39 |
## Example Usage
|
| 40 |
|
|
@@ -84,10 +81,7 @@ for term, probability in zip(terms, probabilities):
|
|
| 84 |
|
| 85 |
## Training Results
|
| 86 |
|
| 87 |
-
**Epochs**: 30
|
| 88 |
-
|
| 89 |
-
**Test
|
| 90 |
-
|
| 91 |
-
**Test Precision**: 0.77
|
| 92 |
-
|
| 93 |
-
**Test Recall**: 0.53
|
|
|
|
| 26 |
|
| 27 |
## Model Specs
|
| 28 |
|
| 29 |
+
- **Vocabulary Size**: 33
|
| 30 |
+
- **Embedding Dimensions**: 480
|
| 31 |
+
- **Attention Heads**: 20
|
| 32 |
+
- **Encoder Layers**: 12
|
| 33 |
+
- **Context Length**: 1026
|
| 34 |
+
- **Total Parameters**: 35M
|
|
|
|
|
|
|
|
|
|
| 35 |
|
| 36 |
## Example Usage
|
| 37 |
|
|
|
|
| 81 |
|
| 82 |
## Training Results
|
| 83 |
|
| 84 |
+
- **Epochs**: 30
|
| 85 |
+
- **Test F1**: 0.63
|
| 86 |
+
- **Test Precision**: 0.77
|
| 87 |
+
- **Test Recall**: 0.53
|
|
|
|
|
|
|
|
|