Initial version
Browse files- README.md +98 -0
- generation_config.json +10 -0
README.md
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: transformers
|
| 3 |
+
license: mit
|
| 4 |
+
base_model: seyonec/ChemBERTa-zinc-base-v1
|
| 5 |
+
tags:
|
| 6 |
+
- PROTAC
|
| 7 |
+
- cheminformatics
|
| 8 |
+
- generated_from_trainer
|
| 9 |
+
model-index:
|
| 10 |
+
- name: ailab-bio/PROTAC-Splitter-EncoderDecoder-lr_cosine_restarts-opt25-rand-smiles
|
| 11 |
+
results: []
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 15 |
+
should probably proofread and complete it, then remove this comment. -->
|
| 16 |
+
|
| 17 |
+
# ailab-bio/PROTAC-Splitter-EncoderDecoder-lr_cosine_restarts-opt25-rand-smiles
|
| 18 |
+
|
| 19 |
+
This model is a fine-tuned version of [seyonec/ChemBERTa-zinc-base-v1](https://huggingface.co/seyonec/ChemBERTa-zinc-base-v1) on the ailab-bio/PROTAC-Splitter-Dataset dataset.
|
| 20 |
+
It achieves the following results on the evaluation set:
|
| 21 |
+
- Loss: 0.3093
|
| 22 |
+
- Linker Heavy Atoms Difference: 0.3372
|
| 23 |
+
- Has Three Substructures: 0.9991
|
| 24 |
+
- Poi Valid: 0.9534
|
| 25 |
+
- Linker Graph Edit Distance Norm: inf
|
| 26 |
+
- Reassembly: 0.5402
|
| 27 |
+
- Valid: 0.9458
|
| 28 |
+
- Poi Heavy Atoms Difference: 1.3755
|
| 29 |
+
- E3 Has Attachment Point(s): 0.9922
|
| 30 |
+
- Poi Heavy Atoms Difference Norm: 0.0422
|
| 31 |
+
- Heavy Atoms Difference: 4.5184
|
| 32 |
+
- Has All Attachment Points: 0.9900
|
| 33 |
+
- Poi Tanimoto Similarity: 0.0
|
| 34 |
+
- Heavy Atoms Difference Norm: 0.0596
|
| 35 |
+
- E3 Heavy Atoms Difference: 0.2046
|
| 36 |
+
- All Ligands Equal: 0.5318
|
| 37 |
+
- E3 Heavy Atoms Difference Norm: -0.0014
|
| 38 |
+
- Linker Heavy Atoms Difference Norm: 0.0044
|
| 39 |
+
- Tanimoto Similarity: 0.0
|
| 40 |
+
- E3 Graph Edit Distance Norm: inf
|
| 41 |
+
- Poi Equal: 0.7580
|
| 42 |
+
- Poi Has Attachment Point(s): 0.9534
|
| 43 |
+
- Poi Graph Edit Distance Norm: inf
|
| 44 |
+
- Num Fragments: 2.9998
|
| 45 |
+
- Linker Graph Edit Distance: inf
|
| 46 |
+
- Linker Valid: 0.9980
|
| 47 |
+
- Linker Tanimoto Similarity: 0.0
|
| 48 |
+
- Poi Graph Edit Distance: inf
|
| 49 |
+
- E3 Equal: 0.8017
|
| 50 |
+
- Linker Has Attachment Point(s): 0.9980
|
| 51 |
+
- Linker Equal: 0.7645
|
| 52 |
+
- E3 Tanimoto Similarity: 0.0
|
| 53 |
+
- Reassembly Nostereo: 0.5728
|
| 54 |
+
- E3 Valid: 0.9922
|
| 55 |
+
- E3 Graph Edit Distance: inf
|
| 56 |
+
|
| 57 |
+
## Model description
|
| 58 |
+
|
| 59 |
+
More information needed
|
| 60 |
+
|
| 61 |
+
## Intended uses & limitations
|
| 62 |
+
|
| 63 |
+
More information needed
|
| 64 |
+
|
| 65 |
+
## Training and evaluation data
|
| 66 |
+
|
| 67 |
+
More information needed
|
| 68 |
+
|
| 69 |
+
## Training procedure
|
| 70 |
+
|
| 71 |
+
### Training hyperparameters
|
| 72 |
+
|
| 73 |
+
The following hyperparameters were used during training:
|
| 74 |
+
- learning_rate: 5e-05
|
| 75 |
+
- train_batch_size: 128
|
| 76 |
+
- eval_batch_size: 64
|
| 77 |
+
- seed: 42
|
| 78 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 79 |
+
- lr_scheduler_type: cosine_with_restarts
|
| 80 |
+
- lr_scheduler_warmup_steps: 100
|
| 81 |
+
- training_steps: 10000
|
| 82 |
+
- mixed_precision_training: Native AMP
|
| 83 |
+
|
| 84 |
+
### Training results
|
| 85 |
+
|
| 86 |
+
| Training Loss | Epoch | Step | Validation Loss | Linker Heavy Atoms Difference | Has Three Substructures | Poi Valid | Linker Graph Edit Distance Norm | Reassembly | Valid | Poi Heavy Atoms Difference | E3 Has Attachment Point(s) | Poi Heavy Atoms Difference Norm | Heavy Atoms Difference | Has All Attachment Points | Poi Tanimoto Similarity | Heavy Atoms Difference Norm | E3 Heavy Atoms Difference | All Ligands Equal | E3 Heavy Atoms Difference Norm | Linker Heavy Atoms Difference Norm | Tanimoto Similarity | E3 Graph Edit Distance Norm | Poi Equal | Poi Has Attachment Point(s) | Poi Graph Edit Distance Norm | Num Fragments | Linker Graph Edit Distance | Linker Valid | Linker Tanimoto Similarity | Poi Graph Edit Distance | E3 Equal | Linker Has Attachment Point(s) | Linker Equal | E3 Tanimoto Similarity | Reassembly Nostereo | E3 Valid | E3 Graph Edit Distance |
|
| 87 |
+
|:-------------:|:------:|:-----:|:---------------:|:-----------------------------:|:-----------------------:|:---------:|:-------------------------------:|:----------:|:------:|:--------------------------:|:--------------------------:|:-------------------------------:|:----------------------:|:-------------------------:|:-----------------------:|:---------------------------:|:-------------------------:|:-----------------:|:------------------------------:|:----------------------------------:|:-------------------:|:---------------------------:|:---------:|:---------------------------:|:----------------------------:|:-------------:|:-------------------------------------------------------------------:|:------------:|:--------------------------:|:-----------------------:|:--------:|:------------------------------:|:------------:|:----------------------:|:-------------------:|:--------:|:----------------------:|
|
| 88 |
+
| 0.0212 | 0.4932 | 5000 | 0.2852 | 0.5099 | 0.9994 | 0.9528 | inf | 0.4464 | 0.9500 | 1.4706 | 0.9979 | 0.0414 | 4.6065 | 0.9903 | 0.0 | 0.0591 | 0.1237 | 0.4396 | -0.0043 | 0.0050 | 0.0 | inf | 0.7208 | 0.9528 | inf | 3.0001 | 24787535410764872823308202666266910179524056896382338942894080.0000 | 0.9975 | 0.0 | inf | 0.7734 | 0.9975 | 0.6416 | 0.0 | 0.4783 | 0.9979 | inf |
|
| 89 |
+
| 0.011 | 0.7398 | 7500 | 0.3022 | 0.3255 | 0.9990 | 0.9389 | 0.0511 | 0.4896 | 0.9294 | 2.2561 | 0.9905 | 0.0673 | 6.6114 | 0.9858 | 0.0 | 0.0846 | 0.5439 | 0.4827 | 0.0117 | -0.0008 | 0.0 | inf | 0.7332 | 0.9389 | inf | 3.0001 | 27443342776203963485595838968590031296472552664861602492710912.0000 | 0.9973 | 0.0 | inf | 0.7827 | 0.9973 | 0.7075 | 0.0 | 0.5176 | 0.9905 | inf |
|
| 90 |
+
| 0.007 | 0.9864 | 10000 | 0.3093 | 0.3372 | 0.9991 | 0.9534 | inf | 0.5402 | 0.9458 | 1.3755 | 0.9922 | 0.0422 | 4.5184 | 0.9900 | 0.0 | 0.0596 | 0.2046 | 0.5318 | -0.0014 | 0.0044 | 0.0 | inf | 0.7580 | 0.9534 | inf | 2.9998 | inf | 0.9980 | 0.0 | inf | 0.8017 | 0.9980 | 0.7645 | 0.0 | 0.5728 | 0.9922 | inf |
|
| 91 |
+
|
| 92 |
+
|
| 93 |
+
### Framework versions
|
| 94 |
+
|
| 95 |
+
- Transformers 4.44.2
|
| 96 |
+
- Pytorch 2.4.1+cu121
|
| 97 |
+
- Datasets 3.0.0
|
| 98 |
+
- Tokenizers 0.19.1
|
generation_config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token_id": 0,
|
| 3 |
+
"do_sample": true,
|
| 4 |
+
"eos_token_id": 2,
|
| 5 |
+
"max_length": 512,
|
| 6 |
+
"num_beams": 5,
|
| 7 |
+
"pad_token_id": 1,
|
| 8 |
+
"top_k": 20,
|
| 9 |
+
"transformers_version": "4.44.2"
|
| 10 |
+
}
|