nepp1d0
/

prot_bert_classification_finetuned

Text Classification

Generated from Trainer

Model card Files Files and versions

Metrics Training metrics Community

nepp1d0 commited on May 9, 2022

Commit

7806d1d

·

1 Parent(s): 5be3cf1

update model card README.md

Files changed (1) hide show

README.md +13 -13

README.md CHANGED Viewed

@@ -18,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [nepp1d0/prot_bert-finetuned-smiles-bindingDB](https://huggingface.co/nepp1d0/prot_bert-finetuned-smiles-bindingDB) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6797
-- Accuracy: 0.53
-- F1: 0.2540
-- Precision: 0.6154
-- Recall: 0.16
 ## Model description
@@ -55,14 +55,14 @@ The following hyperparameters were used during training:
 ### Training results
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.6902        | 1.0   | 25   | 0.6897          | 0.53     | 0.1455 | 0.8       | 0.08   |
-| 0.6717        | 2.0   | 50   | 0.6867          | 0.54     | 0.1786 | 0.8333    | 0.1    |
-| 0.6803        | 3.0   | 75   | 0.6846          | 0.55     | 0.2373 | 0.7778    | 0.14   |
-| 0.6911        | 4.0   | 100  | 0.6792          | 0.52     | 0.2258 | 0.5833    | 0.14   |
-| 0.6447        | 5.0   | 125  | 0.6799          | 0.53     | 0.2540 | 0.6154    | 0.16   |
-| 0.6287        | 6.0   | 150  | 0.6797          | 0.53     | 0.2540 | 0.6154    | 0.16   |
 ### Framework versions

 This model is a fine-tuned version of [nepp1d0/prot_bert-finetuned-smiles-bindingDB](https://huggingface.co/nepp1d0/prot_bert-finetuned-smiles-bindingDB) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.5675
+- Accuracy: 0.7299
+- F1: 0.7377
+- Precision: 0.6995
+- Recall: 0.7803
 ## Model description
 ### Training results
+| Training Loss | Epoch | Step  | Validation Loss | Accuracy | F1     | Precision | Recall |
+|:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| 0.4221        | 1.0   | 3332  | 0.6152          | 0.6615   | 0.6711 | 0.6367    | 0.7093 |
+| 0.4133        | 2.0   | 6664  | 0.5840          | 0.6845   | 0.6718 | 0.6805    | 0.6634 |
+| 0.4293        | 3.0   | 9996  | 0.5727          | 0.7116   | 0.7094 | 0.6961    | 0.7232 |
+| 0.3098        | 4.0   | 13328 | 0.5636          | 0.7163   | 0.7220 | 0.6904    | 0.7566 |
+| 0.3881        | 5.0   | 16660 | 0.5629          | 0.7265   | 0.7377 | 0.6918    | 0.7900 |
+| 0.4943        | 6.0   | 19992 | 0.5675          | 0.7299   | 0.7377 | 0.6995    | 0.7803 |
 ### Framework versions