nroggendorff commited on
Commit
247ffde
1 Parent(s): 284b82d

End of training

Browse files
README.md CHANGED
@@ -1,35 +1,55 @@
1
  ---
2
- license: mit
3
- base_model: distilbert/distilbert-base-uncased
4
- metrics:
5
- - accuracy
6
  model-index:
7
- - name: birthday-detector
8
- results: []
9
- datasets:
10
- - nroggendorff/doug
11
- language:
12
- - en
13
- pipeline_tag: text-classification
14
  ---
15
 
16
- # Birthday Detector
 
17
 
18
- This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased).
19
- It achieves the following results on the evaluation set:
 
 
20
 
21
- - Loss: 0.0101
22
- - Accuracy: 1.0
23
 
24
- You can easily run it with inference, or run it in python with:
25
 
26
- ```
27
- from transformers import pipeline
28
- classifier = pipeline("sentiment-analysis", model="nroggendorff/birthday-detector")
29
 
30
- isBirthday = classifier("happy birthday doug")[0]["label"] == "POSITIVE"
31
- ```
32
 
33
- Make sure you have the necessary dependencies installed.
34
 
35
- [This models inspiration](https://youtu.be/Q6fjwHPVqjQ)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: apache-2.0
3
+ base_model: distilbert-base-uncased
4
+ tags:
5
+ - generated_from_trainer
6
  model-index:
7
+ - name: birthday-detector
8
+ results: []
 
 
 
 
 
9
  ---
10
 
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
 
14
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/noaroggendorff/huggingface/runs/x9vqca8a)
15
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/noaroggendorff/huggingface/runs/x9vqca8a)
16
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/noaroggendorff/huggingface/runs/x9vqca8a)
17
+ # birthday-detector
18
 
19
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 
20
 
21
+ ## Model description
22
 
23
+ More information needed
 
 
24
 
25
+ ## Intended uses & limitations
 
26
 
27
+ More information needed
28
 
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 0.0001
39
+ - train_batch_size: 2048
40
+ - eval_batch_size: 32
41
+ - seed: 42
42
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
+ - lr_scheduler_type: linear
44
+ - num_epochs: 32
45
+
46
+ ### Training results
47
+
48
+
49
+
50
+ ### Framework versions
51
+
52
+ - Transformers 4.42.3
53
+ - Pytorch 2.1.2
54
+ - Datasets 2.20.0
55
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:01a1662d5b82e4880a0053fdf92e1213cb27f27b9ea3676e7b586d1958f10062
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:510f13a3120f51c5cb48dbaad31caf9b0bbeeb02c1b57028e1788d591bdc227a
3
  size 267832560
runs/Jul24_22-52-28_6a9606518f52/events.out.tfevents.1721861549.6a9606518f52.34.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32e6c5033624c189015329680583195026dd33269eeb6054e2078c9d7e1c5f1a
3
+ size 5283
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0373c9fe5cebb2420f40972dda1c4eb842ec433a06b9ca00532df5cd975d46e8
3
- size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:232fc6f2258d5ff25ffdc82c90f84e203ae5ae9d0710972bd2ae43c63c905ab8
3
+ size 5112