Hugging Face's logo

ml6team
/
distilbert-base-german-cased-toxic-comments

Text Classification
Transformers
PyTorch
German
distilbert
german
classification
Inference Endpoints
Model card Files Community
1
Konstantin commited on
Commit
013b41c
•
1 Parent(s): 04939ef

add model

Browse files
Files changed (2) hide show
  1. config.json +10 -2
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "drive/My Drive/Colab Notebooks/Toxicity Detection/models/results/checkpoint-2000",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForSequenceClassification"
@@ -8,7 +8,15 @@
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
 
 
 
 
11
  "initializer_range": 0.02,
 
 
 
 
12
  "max_position_embeddings": 512,
13
  "model_type": "distilbert",
14
  "n_heads": 12,
@@ -21,6 +29,6 @@
21
  "sinusoidal_pos_embds": true,
22
  "tie_weights_": true,
23
  "torch_dtype": "float32",
24
- "transformers_version": "4.12.5",
25
  "vocab_size": 31102
26
  }
 
1
  {
2
+ "_name_or_path": "distilbert-base-german-cased",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForSequenceClassification"
 
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "non_toxic",
13
+ "1": "toxic"
14
+ },
15
  "initializer_range": 0.02,
16
+ "label2id": {
17
+ "non_toxic": 0,
18
+ "toxic": 1
19
+ },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",
22
  "n_heads": 12,
 
29
  "sinusoidal_pos_embds": true,
30
  "tie_weights_": true,
31
  "torch_dtype": "float32",
32
+ "transformers_version": "4.15.0",
33
  "vocab_size": 31102
34
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5122d307e21577e6065c618760778c9779f3d0b9d51c8fe65daba30ad642e632
3
  size 269640433
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97ed07ae0eff54781e04cb15d8c3d0d309d39f4eebb476e0124f060e20473ff5
3
  size 269640433