Create configuration_textcnn.py

configuration_textcnn.py

@@ -0,0 +1,33 @@
+from transformers.configuration_utils import PretrainedConfig
+
+
+class TextCNNConfig(PretrainedConfig):
+    def __init__(
+        self,
+        vocab_size=30000,
+        embed_dim=300,
+        filter_sizes=[1,2,3,4,5],
+        num_filters=[128]*5,
+        dropout=0.5,
+        num_labels=2,
+        id2label={0:"standard", 1:"dialect"},
+        label2id={"standard":0, "dialect":1},
+        bos_token_id=0,
+        eos_token_id=1,
+        pad_token_id=3,
+        **kwargs,
+    ):
+        super().__init__(
+            num_labels=num_labels,
+            id2label=id2label,
+            label2id=label2id,
+            bos_token_id=bos_token_id,
+            eos_token_id=eos_token_id,
+            pad_token_id=pad_token_id,
+            **kwargs,
+        )
+        self.vocab_size = vocab_size
+        self.embed_dim = embed_dim
+        self.filter_sizes = filter_sizes
+        self.num_filters = num_filters
+        self.dropout = dropout