Hugging Face's logo

qunfengd
/
esm2_t12_35M_UR50D-finetuned-cytosol-membrane-classification

Text Classification
Transformers
TensorFlow
esm
generated_from_keras_callback
Inference Endpoints
Model card Files Community
qunfengd commited on
Commit
cf6d611
•
1 Parent(s): a6db033

Upload TFEsmForSequenceClassification

Browse files
Files changed (3) hide show
  1. README.md +9 -9
  2. config.json +1 -1
  3. tf_model.h5 +1 -1
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  license: mit
 
3
  tags:
4
  - generated_from_keras_callback
5
- base_model: facebook/esm2_t12_35M_UR50D
6
  model-index:
7
  - name: esm2_t12_35M_UR50D-finetuned-cytosol-membrane-classification
8
  results: []
@@ -15,10 +15,10 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [facebook/esm2_t12_35M_UR50D](https://huggingface.co/facebook/esm2_t12_35M_UR50D) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.1091
19
- - Train Accuracy: 0.9629
20
- - Validation Loss: 0.1706
21
- - Validation Accuracy: 0.9457
22
  - Epoch: 2
23
 
24
  ## Model description
@@ -45,14 +45,14 @@ The following hyperparameters were used during training:
45
 
46
  | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
47
  |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
48
- | 0.2430 | 0.9263 | 0.2047 | 0.9363 | 0 |
49
- | 0.1503 | 0.9512 | 0.1754 | 0.9371 | 1 |
50
- | 0.1091 | 0.9629 | 0.1706 | 0.9457 | 2 |
51
 
52
 
53
  ### Framework versions
54
 
55
- - Transformers 4.40.0
56
  - TensorFlow 2.15.0
57
  - Datasets 2.19.0
58
  - Tokenizers 0.19.1
 
1
  ---
2
  license: mit
3
+ base_model: facebook/esm2_t12_35M_UR50D
4
  tags:
5
  - generated_from_keras_callback
 
6
  model-index:
7
  - name: esm2_t12_35M_UR50D-finetuned-cytosol-membrane-classification
8
  results: []
 
15
 
16
  This model is a fine-tuned version of [facebook/esm2_t12_35M_UR50D](https://huggingface.co/facebook/esm2_t12_35M_UR50D) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.0918
19
+ - Train Accuracy: 0.9694
20
+ - Validation Loss: 0.1640
21
+ - Validation Accuracy: 0.9495
22
  - Epoch: 2
23
 
24
  ## Model description
 
45
 
46
  | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
47
  |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
48
+ | 0.2427 | 0.9271 | 0.1663 | 0.9464 | 0 |
49
+ | 0.1468 | 0.9538 | 0.1505 | 0.9495 | 1 |
50
+ | 0.0918 | 0.9694 | 0.1640 | 0.9495 | 2 |
51
 
52
 
53
  ### Framework versions
54
 
55
+ - Transformers 4.40.1
56
  - TensorFlow 2.15.0
57
  - Datasets 2.19.0
58
  - Tokenizers 0.19.1
config.json CHANGED
@@ -23,7 +23,7 @@
23
  "position_embedding_type": "rotary",
24
  "token_dropout": true,
25
  "torch_dtype": "float32",
26
- "transformers_version": "4.40.0",
27
  "use_cache": true,
28
  "vocab_list": null,
29
  "vocab_size": 33
 
23
  "position_embedding_type": "rotary",
24
  "token_dropout": true,
25
  "torch_dtype": "float32",
26
+ "transformers_version": "4.40.1",
27
  "use_cache": true,
28
  "vocab_list": null,
29
  "vocab_size": 33
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aa506f2418f7d88617963eb6b1602cf168a9abd8f72a6a8e262e218ef123f793
3
  size 136284336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd6cbe11cb9858c0136202511e1aa6afe98262386f82a9e1f84c3dc6b13a3d78
3
  size 136284336