soonbob commited on
Commit
212cdca
·
verified ·
1 Parent(s): e95c269

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +46 -17
README.md CHANGED
@@ -2,12 +2,28 @@
2
  library_name: transformers
3
  tags:
4
  - generated_from_trainer
 
 
 
5
  metrics:
6
  - accuracy
7
  - f1
8
  model-index:
9
- - name: hf_transformer
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  license: mit
12
  datasets:
13
  - nyu-mll/glue
@@ -21,29 +37,42 @@ pipeline_tag: text-classification
21
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
22
  should probably proofread and complete it, then remove this comment. -->
23
 
24
- # hf_transformer
25
 
26
- This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
27
- It achieves the following results on the evaluation set:
28
- - Loss: 0.8276
29
- - Accuracy: 0.6368
30
- - F1: 0.6358
31
 
32
- ## Model description
 
 
 
33
 
34
- More information needed
35
 
36
- ## Intended uses & limitations
 
 
 
37
 
38
- More information needed
39
 
40
- ## Training and evaluation data
 
 
 
 
 
41
 
42
- More information needed
 
 
 
 
 
43
 
44
- ## Training procedure
45
 
46
- ### Training hyperparameters
47
 
48
  The following hyperparameters were used during training:
49
  - learning_rate: 0.001
@@ -54,7 +83,7 @@ The following hyperparameters were used during training:
54
  - lr_scheduler_type: linear
55
  - num_epochs: 3
56
 
57
- ### Training results
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
60
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
 
2
  library_name: transformers
3
  tags:
4
  - generated_from_trainer
5
+ - mnli
6
+ - text-classification
7
+ - bert
8
  metrics:
9
  - accuracy
10
  - f1
11
  model-index:
12
+ - name: mnli-finetuned-bert-base-cased
13
+ results:
14
+ - task:
15
+ type: text-classification
16
+ name: Natural Language Inference
17
+ dataset:
18
+ name: MultiNLI
19
+ type: nyu-mll/glue
20
+ metrics:
21
+ - name: Accuracy
22
+ type: accuracy
23
+ value: 0.6368
24
+ - name: F1
25
+ type: f1
26
+ value: 0.6358
27
  license: mit
28
  datasets:
29
  - nyu-mll/glue
 
37
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
38
  should probably proofread and complete it, then remove this comment. -->
39
 
40
+ # soonbob/mnli-finetuned-bert-base-cased
41
 
42
+ MNLI 데이터셋을 학습시킨 BERT
43
+ 파인튜닝 연습용으로 만든 것입니다.
44
+
45
+ This is a BERT-based model fine-tuned on the [Multi-Genre Natural Language Inference (MultiNLI)](https://huggingface.co/datasets/glue/viewer/mnli) dataset for the task of **natural language inference** (NLI), using Hugging Face's `Trainer`.
 
46
 
47
+ It classifies a pair of sentences into one of the following classes:
48
+ - **entailment**
49
+ - **neutral**
50
+ - **contradiction**
51
 
52
+ ## 🧠 Intended Use
53
 
54
+ This model can be used for:
55
+ - Evaluating whether one sentence logically follows from another
56
+ - Sentence-pair classification tasks
57
+ - Transfer learning for other NLI-style problems
58
 
 
59
 
60
+ It achieves the following results on the evaluation set:
61
+ - Loss: 0.8276
62
+ - Accuracy: 0.6368
63
+ - F1: 0.6358
64
+
65
+ ## ⚙️ Training Details
66
 
67
+ - Base model: [`bert-base-cased`](https://huggingface.co/bert-base-cased)
68
+ - Dataset: `nyu-mll/glue`, subset: `mnli`
69
+ - Epochs: 3
70
+ - Learning rate: 1e-3
71
+ - Optimizer: AdamW
72
+ - Scheduler: Linear
73
 
 
74
 
75
+ ### 🏋️ Training hyperparameters
76
 
77
  The following hyperparameters were used during training:
78
  - learning_rate: 0.001
 
83
  - lr_scheduler_type: linear
84
  - num_epochs: 3
85
 
86
+ ### 🏋️ Training Logs
87
 
88
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
89
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|