dancingninjas commited on
Commit
95556c7
1 Parent(s): 6d7eed6

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -5
config.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "_name_or_path": "distilbert-base-uncased",
3
  "activation": "gelu",
4
  "architectures": [
5
- "TFAutoModelForSequenceClassification"
6
  ],
7
  "attention_dropout": 0.1,
8
  "dim": 768,
@@ -10,12 +10,19 @@
10
  "hidden_dim": 3072,
11
  "initializer_range": 0.02,
12
  "max_position_embeddings": 512,
13
- "model_type": "distilbert",
14
  "n_heads": 12,
15
  "n_layers": 6,
16
  "pad_token_id": 0,
17
  "problem_type": "single_label_classification",
18
  "seq_classif_dropout": 0.2,
19
  "transformers_version": "4.26.1",
20
- "vocab_size": 30522
21
- }
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "dancingninjas/sentiment-model",
3
  "activation": "gelu",
4
  "architectures": [
5
+ "TFAutoModelForSequenceClassification"
6
  ],
7
  "attention_dropout": 0.1,
8
  "dim": 768,
 
10
  "hidden_dim": 3072,
11
  "initializer_range": 0.02,
12
  "max_position_embeddings": 512,
13
+ "model_type": "tf_model",
14
  "n_heads": 12,
15
  "n_layers": 6,
16
  "pad_token_id": 0,
17
  "problem_type": "single_label_classification",
18
  "seq_classif_dropout": 0.2,
19
  "transformers_version": "4.26.1",
20
+ "vocab_size": 30522,
21
+ "description": "A fine-tuned DistilBERT model for sentiment analysis.",
22
+ "tags": ["text-classification", "sentiment-analysis"],
23
+ "pipeline_tag": "text-classification",
24
+ "framework": "transformers",
25
+ "pytorch_version": "1.8.1",
26
+ "tensorflow_version": "2.7.0",
27
+ "license": "MIT"
28
+ }