Commit
·
9a0b5b4
1
Parent(s):
3b80f5d
Delete configuration_gcn.py
Browse files- configuration_gcn.py +0 -33
configuration_gcn.py
DELETED
@@ -1,33 +0,0 @@
|
|
1 |
-
from transformers import PretrainedConfig
|
2 |
-
from typing import List
|
3 |
-
class GCNConfig(PretrainedConfig):
|
4 |
-
model_type = "gcn"
|
5 |
-
|
6 |
-
def __init__(
|
7 |
-
self,
|
8 |
-
input_feature: int=64,
|
9 |
-
emb_input: int=20,
|
10 |
-
hidden_size: int=64,
|
11 |
-
n_layers: int=6,
|
12 |
-
num_classes: int=1,
|
13 |
-
|
14 |
-
smiles: List[str] = None,
|
15 |
-
processor_class: str = "SmilesProcessor",
|
16 |
-
**kwargs,
|
17 |
-
):
|
18 |
-
|
19 |
-
self.input_feature = input_feature # the dimension of input feature
|
20 |
-
self.emb_input = emb_input # the embedding dimension of input feature
|
21 |
-
self.hidden_size = hidden_size # the hidden size of GCN
|
22 |
-
self.n_layers = n_layers # the number of GCN layers
|
23 |
-
self.num_classes = num_classes # the number of output classes
|
24 |
-
|
25 |
-
self.smiles = smiles # process smiles
|
26 |
-
self.processor_class = processor_class
|
27 |
-
|
28 |
-
super().__init__(**kwargs)
|
29 |
-
|
30 |
-
|
31 |
-
if __name__ == "__main__":
|
32 |
-
gcn_config = GCNConfig(input_feature=64, emb_input=20, hidden_size=64, n_layers=6, num_classes=1, smiles=["C", "CC", "CCC"], processor_class="SmilesProcessor")
|
33 |
-
gcn_config.save_pretrained("custom-gcn")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|