File size: 933 Bytes
556d5b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 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 |