Huhujingjing commited on
Commit
9a0b5b4
·
1 Parent(s): 3b80f5d

Delete configuration_gcn.py

Browse files
Files changed (1) hide show
  1. 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")