nepp1d0
/

prot_bert-finetuned-smiles-bindingDB

Generated from Trainer

Model card Files Files and versions Metrics Training metrics Community

nepp1d0 commited on May 5, 2022

Commit

a5016fb

·

1 Parent(s): 292f78c

update model card README.md

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [Rostlab/prot_bert](https://huggingface.co/Rostlab/prot_bert) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.8420
 ## Model description
@@ -32,7 +32,7 @@ More information needed
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- learning_rate: 1e-07
 - train_batch_size: 1
 - eval_batch_size: 1
 - seed: 42
@@ -45,9 +45,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step  | Validation Loss |
 |:-------------:|:-----:|:-----:|:---------------:|
-| 2.3927        | 1.0   | 10000 | 2.0457          |
-| 1.9424        | 2.0   | 20000 | 1.8816          |
-| 1.8575        | 3.0   | 30000 | 1.8551          |
 ### Framework versions

 This model is a fine-tuned version of [Rostlab/prot_bert](https://huggingface.co/Rostlab/prot_bert) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 1.2121
 ## Model description
 ### Training hyperparameters
 The following hyperparameters were used during training:
+- learning_rate: 1e-06
 - train_batch_size: 1
 - eval_batch_size: 1
 - seed: 42
 | Training Loss | Epoch | Step  | Validation Loss |
 |:-------------:|:-----:|:-----:|:---------------:|
+| 1.6742        | 1.0   | 10000 | 1.3855          |
+| 1.3228        | 2.0   | 20000 | 1.2549          |
+| 1.2384        | 3.0   | 30000 | 1.2114          |
 ### Framework versions