ribesstefano commited on
Commit
d8becbf
·
verified ·
1 Parent(s): 2d888c2

Initial version

Browse files
Files changed (2) hide show
  1. README.md +97 -0
  2. generation_config.json +10 -0
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.25
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.25
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.3624
22
+ - Linker Equal: 0.8351
23
+ - E3 Graph Edit Distance Norm: inf
24
+ - Poi Heavy Atoms Difference Norm: 0.0311
25
+ - Poi Has Attachment Point(s): 0.9593
26
+ - Poi Graph Edit Distance Norm: inf
27
+ - Valid: 0.9367
28
+ - Poi Valid: 0.9593
29
+ - Poi Graph Edit Distance: inf
30
+ - Reassembly Nostereo: 0.6220
31
+ - Linker Has Attachment Point(s): 0.9970
32
+ - Poi Heavy Atoms Difference: 1.1101
33
+ - Has Three Substructures: 1.0
34
+ - E3 Heavy Atoms Difference Norm: 0.0153
35
+ - Linker Valid: 0.9970
36
+ - Linker Tanimoto Similarity: 0.0
37
+ - E3 Has Attachment Point(s): 0.9798
38
+ - Poi Equal: 0.7936
39
+ - All Ligands Equal: 0.5870
40
+ - Heavy Atoms Difference Norm: 0.0731
41
+ - Linker Heavy Atoms Difference Norm: 0.0030
42
+ - Heavy Atoms Difference: 5.4708
43
+ - Tanimoto Similarity: 0.0
44
+ - Num Fragments: 3.0
45
+ - E3 Graph Edit Distance: inf
46
+ - Linker Graph Edit Distance: inf
47
+ - Linker Heavy Atoms Difference: 0.2800
48
+ - Poi Tanimoto Similarity: 0.0
49
+ - Linker Graph Edit Distance Norm: inf
50
+ - Has All Attachment Points: 0.9888
51
+ - E3 Tanimoto Similarity: 0.0
52
+ - Reassembly: 0.5944
53
+ - E3 Heavy Atoms Difference: 0.4994
54
+ - E3 Equal: 0.8276
55
+ - E3 Valid: 0.9798
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 | Linker Equal | E3 Graph Edit Distance Norm | Poi Heavy Atoms Difference Norm | Poi Has Attachment Point(s) | Poi Graph Edit Distance Norm | Valid | Poi Valid | Poi Graph Edit Distance | Reassembly Nostereo | Linker Has Attachment Point(s) | Poi Heavy Atoms Difference | Has Three Substructures | E3 Heavy Atoms Difference Norm | Linker Valid | Linker Tanimoto Similarity | E3 Has Attachment Point(s) | Poi Equal | All Ligands Equal | Heavy Atoms Difference Norm | Linker Heavy Atoms Difference Norm | Heavy Atoms Difference | Tanimoto Similarity | Num Fragments | E3 Graph Edit Distance | Linker Graph Edit Distance | Linker Heavy Atoms Difference | Poi Tanimoto Similarity | Linker Graph Edit Distance Norm | Has All Attachment Points | E3 Tanimoto Similarity | Reassembly | E3 Heavy Atoms Difference | E3 Equal | E3 Valid |
87
+ |:-------------:|:-------:|:------:|:---------------:|:------------:|:---------------------------:|:-------------------------------:|:---------------------------:|:----------------------------:|:------:|:---------:|:-----------------------:|:-------------------:|:------------------------------:|:--------------------------:|:-----------------------:|:------------------------------:|:------------:|:--------------------------:|:--------------------------:|:---------:|:-----------------:|:---------------------------:|:----------------------------------:|:----------------------:|:-------------------:|:-------------:|:----------------------:|:--------------------------:|:-----------------------------:|:-----------------------:|:-------------------------------:|:-------------------------:|:----------------------:|:----------:|:-------------------------:|:--------:|:--------:|
88
+ | 0.0006 | 31.5582 | 80000 | 0.3577 | 0.8324 | inf | 0.0305 | 0.9579 | inf | 0.9328 | 0.9579 | inf | 0.6230 | 0.9967 | 1.0626 | 0.9999 | 0.0251 | 0.9967 | 0.0 | 0.9771 | 0.7863 | 0.5856 | 0.0758 | 0.0051 | 5.7523 | 0.0 | 3.0001 | inf | inf | 0.2647 | 0.0 | inf | 0.9890 | 0.0 | 0.5934 | 0.8042 | 0.8272 | 0.9771 |
89
+ | 0.0005 | 39.4477 | 100000 | 0.3624 | 0.8351 | inf | 0.0311 | 0.9593 | inf | 0.9367 | 0.9593 | inf | 0.6220 | 0.9970 | 1.1101 | 1.0 | 0.0153 | 0.9970 | 0.0 | 0.9798 | 0.7936 | 0.5870 | 0.0731 | 0.0030 | 5.4708 | 0.0 | 3.0 | inf | inf | 0.2800 | 0.0 | inf | 0.9888 | 0.0 | 0.5944 | 0.4994 | 0.8276 | 0.9798 |
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
+ }