Model save
Browse files- README.md +63 -0
- config.json +33 -0
- model.safetensors +3 -0
- runs/Mar28_19-30-14_265209550b7f/events.out.tfevents.1711654215.265209550b7f.33.0 +3 -0
- runs/Mar28_19-34-56_265209550b7f/events.out.tfevents.1711654503.265209550b7f.33.1 +3 -0
- runs/Mar28_20-02-13_265209550b7f/events.out.tfevents.1711656137.265209550b7f.33.2 +3 -0
- runs/Mar28_20-04-38_265209550b7f/events.out.tfevents.1711656283.265209550b7f.33.3 +3 -0
- runs/Mar28_20-07-13_265209550b7f/events.out.tfevents.1711656435.265209550b7f.33.4 +3 -0
- runs/Mar28_20-07-47_265209550b7f/events.out.tfevents.1711656470.265209550b7f.33.5 +3 -0
- runs/Mar28_20-36-37_265209550b7f/events.out.tfevents.1711658202.265209550b7f.33.6 +3 -0
- runs/Mar28_21-04-43_265209550b7f/events.out.tfevents.1711659891.265209550b7f.33.7 +3 -0
- runs/Mar28_21-06-10_265209550b7f/events.out.tfevents.1711659971.265209550b7f.33.8 +3 -0
- runs/Mar28_21-24-20_265209550b7f/events.out.tfevents.1711661061.265209550b7f.33.9 +3 -0
- special_tokens_map.json +7 -0
- tokenizer.json +0 -0
- tokenizer_config.json +55 -0
- training_args.bin +3 -0
- vocab.txt +0 -0
README.md
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
base_model: distilbert-base-uncased
|
4 |
+
tags:
|
5 |
+
- generated_from_trainer
|
6 |
+
metrics:
|
7 |
+
- accuracy
|
8 |
+
- f1
|
9 |
+
model-index:
|
10 |
+
- name: distilbert-base-uncased-finetuned-ai-humor
|
11 |
+
results: []
|
12 |
+
---
|
13 |
+
|
14 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
15 |
+
should probably proofread and complete it, then remove this comment. -->
|
16 |
+
|
17 |
+
# distilbert-base-uncased-finetuned-ai-humor
|
18 |
+
|
19 |
+
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
|
20 |
+
It achieves the following results on the evaluation set:
|
21 |
+
- Loss: 0.4763
|
22 |
+
- Accuracy: 0.7714
|
23 |
+
- F1: 0.7715
|
24 |
+
|
25 |
+
## Model description
|
26 |
+
|
27 |
+
More information needed
|
28 |
+
|
29 |
+
## Intended uses & limitations
|
30 |
+
|
31 |
+
More information needed
|
32 |
+
|
33 |
+
## Training and evaluation data
|
34 |
+
|
35 |
+
More information needed
|
36 |
+
|
37 |
+
## Training procedure
|
38 |
+
|
39 |
+
### Training hyperparameters
|
40 |
+
|
41 |
+
The following hyperparameters were used during training:
|
42 |
+
- learning_rate: 2e-05
|
43 |
+
- train_batch_size: 64
|
44 |
+
- eval_batch_size: 64
|
45 |
+
- seed: 42
|
46 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
47 |
+
- lr_scheduler_type: linear
|
48 |
+
- num_epochs: 2
|
49 |
+
|
50 |
+
### Training results
|
51 |
+
|
52 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
|
53 |
+
|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
|
54 |
+
| 0.2993 | 1.0 | 16 | 0.5179 | 0.75 | 0.7490 |
|
55 |
+
| 0.2676 | 2.0 | 32 | 0.4763 | 0.7714 | 0.7715 |
|
56 |
+
|
57 |
+
|
58 |
+
### Framework versions
|
59 |
+
|
60 |
+
- Transformers 4.38.2
|
61 |
+
- Pytorch 2.1.2+cpu
|
62 |
+
- Datasets 2.1.0
|
63 |
+
- Tokenizers 0.15.2
|
config.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "distilbert-base-uncased",
|
3 |
+
"activation": "gelu",
|
4 |
+
"architectures": [
|
5 |
+
"DistilBertForSequenceClassification"
|
6 |
+
],
|
7 |
+
"attention_dropout": 0.1,
|
8 |
+
"dim": 768,
|
9 |
+
"dropout": 0.1,
|
10 |
+
"hidden_dim": 3072,
|
11 |
+
"id2label": {
|
12 |
+
"0": "joke",
|
13 |
+
"1": "non-joke"
|
14 |
+
},
|
15 |
+
"initializer_range": 0.02,
|
16 |
+
"label2id": {
|
17 |
+
"joke": "0",
|
18 |
+
"non-joke": "1"
|
19 |
+
},
|
20 |
+
"max_position_embeddings": 512,
|
21 |
+
"model_type": "distilbert",
|
22 |
+
"n_heads": 12,
|
23 |
+
"n_layers": 6,
|
24 |
+
"pad_token_id": 0,
|
25 |
+
"problem_type": "single_label_classification",
|
26 |
+
"qa_dropout": 0.1,
|
27 |
+
"seq_classif_dropout": 0.2,
|
28 |
+
"sinusoidal_pos_embds": false,
|
29 |
+
"tie_weights_": true,
|
30 |
+
"torch_dtype": "float32",
|
31 |
+
"transformers_version": "4.38.2",
|
32 |
+
"vocab_size": 30522
|
33 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f63dc867bac64392730955fb66c5d9500f26c2d04b93118763be577142ff9e7d
|
3 |
+
size 267832560
|
runs/Mar28_19-30-14_265209550b7f/events.out.tfevents.1711654215.265209550b7f.33.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9b06159dc768db2e29aabe7831075ef0199cb131bd8807fe87b99f39dd73dc29
|
3 |
+
size 4647
|
runs/Mar28_19-34-56_265209550b7f/events.out.tfevents.1711654503.265209550b7f.33.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4923d5174e22c781d201023b03e23c089d91fbc89333098fab15ae221b05a455
|
3 |
+
size 4647
|
runs/Mar28_20-02-13_265209550b7f/events.out.tfevents.1711656137.265209550b7f.33.2
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f81e7b60e9c05642ac179334beff2e1590dcce5e5969058f3a0328cf90c1a775
|
3 |
+
size 4647
|
runs/Mar28_20-04-38_265209550b7f/events.out.tfevents.1711656283.265209550b7f.33.3
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0b30690765df23f7ad96b064d308aa6727561dd2dd4157f1b4826058ef0c6957
|
3 |
+
size 4647
|
runs/Mar28_20-07-13_265209550b7f/events.out.tfevents.1711656435.265209550b7f.33.4
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d4992cb492a20deed44448b0b6e1022195827b4168646008f40ae82113011429
|
3 |
+
size 4647
|
runs/Mar28_20-07-47_265209550b7f/events.out.tfevents.1711656470.265209550b7f.33.5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4c206cd850f7e52c9e662ccf8f209407a6b39c3ba0c04af5fe85a1524a540b2c
|
3 |
+
size 4854
|
runs/Mar28_20-36-37_265209550b7f/events.out.tfevents.1711658202.265209550b7f.33.6
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f625d153b84e454b0801c70ecc385b9f95e14c30c15b5074a66b7b19b5761d60
|
3 |
+
size 4903
|
runs/Mar28_21-04-43_265209550b7f/events.out.tfevents.1711659891.265209550b7f.33.7
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:983ab17aeb4901fb08a76cee50943a942948f91e59cbc8dbd98e96d06c512402
|
3 |
+
size 4696
|
runs/Mar28_21-06-10_265209550b7f/events.out.tfevents.1711659971.265209550b7f.33.8
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1d36117727898b2ff4bd155bc861f8bd76511decb8486f9455423227e7facaa2
|
3 |
+
size 4903
|
runs/Mar28_21-24-20_265209550b7f/events.out.tfevents.1711661061.265209550b7f.33.9
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d538cc0a15f0626746d32f0c20ead079ba66582c35c81c16b23b3ee0af0038f9
|
3 |
+
size 6182
|
special_tokens_map.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"cls_token": "[CLS]",
|
3 |
+
"mask_token": "[MASK]",
|
4 |
+
"pad_token": "[PAD]",
|
5 |
+
"sep_token": "[SEP]",
|
6 |
+
"unk_token": "[UNK]"
|
7 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"added_tokens_decoder": {
|
3 |
+
"0": {
|
4 |
+
"content": "[PAD]",
|
5 |
+
"lstrip": false,
|
6 |
+
"normalized": false,
|
7 |
+
"rstrip": false,
|
8 |
+
"single_word": false,
|
9 |
+
"special": true
|
10 |
+
},
|
11 |
+
"100": {
|
12 |
+
"content": "[UNK]",
|
13 |
+
"lstrip": false,
|
14 |
+
"normalized": false,
|
15 |
+
"rstrip": false,
|
16 |
+
"single_word": false,
|
17 |
+
"special": true
|
18 |
+
},
|
19 |
+
"101": {
|
20 |
+
"content": "[CLS]",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false,
|
25 |
+
"special": true
|
26 |
+
},
|
27 |
+
"102": {
|
28 |
+
"content": "[SEP]",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false,
|
33 |
+
"special": true
|
34 |
+
},
|
35 |
+
"103": {
|
36 |
+
"content": "[MASK]",
|
37 |
+
"lstrip": false,
|
38 |
+
"normalized": false,
|
39 |
+
"rstrip": false,
|
40 |
+
"single_word": false,
|
41 |
+
"special": true
|
42 |
+
}
|
43 |
+
},
|
44 |
+
"clean_up_tokenization_spaces": true,
|
45 |
+
"cls_token": "[CLS]",
|
46 |
+
"do_lower_case": true,
|
47 |
+
"mask_token": "[MASK]",
|
48 |
+
"model_max_length": 512,
|
49 |
+
"pad_token": "[PAD]",
|
50 |
+
"sep_token": "[SEP]",
|
51 |
+
"strip_accents": null,
|
52 |
+
"tokenize_chinese_chars": true,
|
53 |
+
"tokenizer_class": "DistilBertTokenizer",
|
54 |
+
"unk_token": "[UNK]"
|
55 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ba873475c55c0ef5e32474eb73b054702e153fc719db21991ef020f95694708c
|
3 |
+
size 4920
|
vocab.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|