ribesstefano commited on
Commit
ff3c9b9
·
verified ·
1 Parent(s): 4fc83d2

Initial version

Browse files
Files changed (3) hide show
  1. README.md +97 -0
  2. generation_config.json +10 -0
  3. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_reduce-rand-smiles-train-0.5
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_reduce-rand-smiles-train-0.5
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.3751
22
+ - Reassembly: 0.5970
23
+ - E3 Graph Edit Distance Norm: inf
24
+ - Linker Graph Edit Distance Norm: inf
25
+ - All Ligands Equal: 0.5899
26
+ - Poi Heavy Atoms Difference Norm: 0.0419
27
+ - Poi Graph Edit Distance Norm: inf
28
+ - Linker Heavy Atoms Difference: 0.2883
29
+ - Linker Equal: 0.8472
30
+ - Poi Has Attachment Point(s): 0.9510
31
+ - E3 Tanimoto Similarity: 0.0
32
+ - Reassembly Nostereo: 0.6329
33
+ - E3 Graph Edit Distance: inf
34
+ - Linker Has Attachment Point(s): 0.9977
35
+ - E3 Heavy Atoms Difference Norm: 0.0037
36
+ - Linker Valid: 0.9977
37
+ - Poi Equal: 0.7890
38
+ - Linker Graph Edit Distance: 23016997167138810478786188190104988023843767118069314732687360.0000
39
+ - E3 Equal: 0.8240
40
+ - Num Fragments: 3.0003
41
+ - E3 Heavy Atoms Difference: 0.2737
42
+ - Poi Graph Edit Distance: inf
43
+ - Has Three Substructures: 0.9996
44
+ - Heavy Atoms Difference: 4.7685
45
+ - Linker Tanimoto Similarity: 0.0
46
+ - Poi Heavy Atoms Difference: 1.3950
47
+ - E3 Valid: 0.9944
48
+ - Poi Tanimoto Similarity: 0.0
49
+ - Valid: 0.9450
50
+ - Heavy Atoms Difference Norm: 0.0634
51
+ - Tanimoto Similarity: 0.0
52
+ - E3 Has Attachment Point(s): 0.9944
53
+ - Has All Attachment Points: 0.9938
54
+ - Poi Valid: 0.9510
55
+ - Linker Heavy Atoms Difference Norm: 0.0067
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: reduce_lr_on_plateau
80
+ - lr_scheduler_warmup_steps: 400
81
+ - training_steps: 100000
82
+ - mixed_precision_training: Native AMP
83
+
84
+ ### Training results
85
+
86
+ | Training Loss | Epoch | Step | Validation Loss | Reassembly | E3 Graph Edit Distance Norm | Linker Graph Edit Distance Norm | All Ligands Equal | Poi Heavy Atoms Difference Norm | Poi Graph Edit Distance Norm | Linker Heavy Atoms Difference | Linker Equal | Poi Has Attachment Point(s) | E3 Tanimoto Similarity | Reassembly Nostereo | E3 Graph Edit Distance | Linker Has Attachment Point(s) | E3 Heavy Atoms Difference Norm | Linker Valid | Poi Equal | Linker Graph Edit Distance | E3 Equal | Num Fragments | E3 Heavy Atoms Difference | Poi Graph Edit Distance | Has Three Substructures | Heavy Atoms Difference | Linker Tanimoto Similarity | Poi Heavy Atoms Difference | E3 Valid | Poi Tanimoto Similarity | Valid | Heavy Atoms Difference Norm | Tanimoto Similarity | E3 Has Attachment Point(s) | Has All Attachment Points | Poi Valid | Linker Heavy Atoms Difference Norm |
87
+ |:-------------:|:-------:|:------:|:---------------:|:----------:|:---------------------------:|:-------------------------------:|:-----------------:|:-------------------------------:|:----------------------------:|:-----------------------------:|:------------:|:---------------------------:|:----------------------:|:-------------------:|:----------------------:|:------------------------------:|:------------------------------:|:------------:|:---------:|:-------------------------------------------------------------------:|:--------:|:-------------:|:-------------------------:|:-----------------------:|:-----------------------:|:----------------------:|:--------------------------:|:--------------------------:|:--------:|:-----------------------:|:------:|:---------------------------:|:-------------------:|:--------------------------:|:-------------------------:|:---------:|:----------------------------------:|
88
+ | 0.0006 | 15.7822 | 80000 | 0.3655 | 0.5977 | inf | inf | 0.5904 | 0.0477 | inf | 0.1988 | 0.8371 | 0.9528 | 0.0 | 0.6278 | inf | 0.9952 | 0.0221 | 0.9952 | 0.7851 | inf | 0.8250 | 2.9996 | 0.7440 | inf | 0.9994 | 5.9070 | 0.0 | 1.4796 | 0.9796 | 0.0 | 0.9315 | 0.0782 | 0.0 | 0.9796 | 0.9879 | 0.9528 | -0.0014 |
89
+ | 0.0006 | 19.7278 | 100000 | 0.3751 | 0.5970 | inf | inf | 0.5899 | 0.0419 | inf | 0.2883 | 0.8472 | 0.9510 | 0.0 | 0.6329 | inf | 0.9977 | 0.0037 | 0.9977 | 0.7890 | 23016997167138810478786188190104988023843767118069314732687360.0000 | 0.8240 | 3.0003 | 0.2737 | inf | 0.9996 | 4.7685 | 0.0 | 1.3950 | 0.9944 | 0.0 | 0.9450 | 0.0634 | 0.0 | 0.9944 | 0.9938 | 0.9510 | 0.0067 |
90
+
91
+
92
+ ### Framework versions
93
+
94
+ - Transformers 4.44.2
95
+ - Pytorch 2.4.1+cu121
96
+ - Datasets 3.0.0
97
+ - 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
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f18ce4cbd95e0db674acdf88a65d7216d8955eab9350b232d8b41db01b75645
3
  size 409608164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2790709e38d495ed69451e2e98588967f60dac42ae76070227ad36e62492bafe
3
  size 409608164