bert-fa-sentiment-dksf / model_config.yaml
arxyzan's picture
Upload 5 files
d57bcf7
raw
history blame
396 Bytes
name: distilbert_text_classification
config_type: model
task: text_classification
num_labels: 3
id2label:
'0': negative
'1': positive
'2': neutral
activation: gelu
attention_dropout: 0.1
dim: 768
dropout: 0.1
hidden_dim: 3072
initializer_range: 0.02
max_position_embeddings: 512
n_heads: 12
n_layers: 6
output_past: true
pad_token_id: 0
qa_dropout: 0.1
tie_weights_: true
vocab_size: 42000