omaressamrme commited on
Commit
6d28879
·
verified ·
1 Parent(s): fa43e95

End of training

Browse files
Files changed (4) hide show
  1. README.md +44 -59
  2. SPACE_README.md +1 -1
  3. runtime.txt +1 -0
  4. training_args.bin +1 -1
README.md CHANGED
@@ -1,73 +1,58 @@
1
  ---
2
- language:
3
- - en
4
  license: apache-2.0
 
5
  tags:
6
- - text-classification
7
- - sentiment-analysis
8
- - imdb
9
- - distilbert
10
- pipeline_tag: text-classification
11
- library_name: transformers
12
- datasets:
13
- - imdb
14
- widget:
15
- - text: I absolutely loved this movie!
16
- - text: This was a terrible film. I want my time back.
17
  model-index:
18
- - name: omaressamrme/tuning
19
- results:
20
- - task:
21
- type: text-classification
22
- name: Sentiment Analysis
23
- dataset:
24
- name: IMDb
25
- type: imdb
26
- split: test
27
- metrics:
28
- - type: accuracy
29
- value: 0.31
30
- - type: f1
31
- value: 0.0
32
  ---
33
 
 
 
34
 
35
- # omaressamrme/tuning
36
 
37
- Fine-tuned DistilBERT for sentiment analysis on the IMDb dataset.
 
 
 
 
 
 
 
 
 
38
 
39
- ## Training setup
40
- - Base model: distilbert-base-uncased
41
- - Dataset: IMDb (train/test)
42
- - Epochs: 1
43
- - Learning rate: 2e-05
44
- - Train batch size: 16
45
- - Eval batch size: 32
46
- - Max train samples: 1000
47
- - Max eval samples: 500
48
 
49
- ## Evaluation (test split)
50
- - Accuracy: 0.31
51
- - F1 (binary): 0.0
52
 
53
- ## Usage
54
- ```python
55
- from transformers import pipeline
56
- clf = pipeline("text-classification", model="omaressamrme/tuning")
57
- print(clf("I absolutely loved this movie!"))
58
- ```
59
 
60
- ## Batch inference
61
- You can batch texts using the pipeline:
62
- ```python
63
- texts = ["Great film!", "Worst plot ever."]
64
- preds = clf(texts)
65
- ```
66
 
67
- ## Intended uses & limitations
68
- - Intended for educational/demo sentiment classification.
69
- - Trained on a subset of IMDb for speed; performance is lower than full training.
70
- - May reflect dataset biases; do not use for critical decisions.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
71
 
72
- ## Reproducibility
73
- See the training script in the associated GitHub repo.
 
 
 
1
  ---
2
+ library_name: transformers
 
3
  license: apache-2.0
4
+ base_model: distilbert-base-uncased
5
  tags:
6
+ - generated_from_trainer
 
 
 
 
 
 
 
 
 
 
7
  model-index:
8
+ - name: tuning
9
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
 
15
+ # tuning
16
 
17
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - eval_loss: 0.7005
20
+ - eval_model_preparation_time: 0.0015
21
+ - eval_accuracy: 0.31
22
+ - eval_f1: 0.0
23
+ - eval_runtime: 27.5721
24
+ - eval_samples_per_second: 18.134
25
+ - eval_steps_per_second: 0.58
26
+ - step: 0
27
 
28
+ ## Model description
 
 
 
 
 
 
 
 
29
 
30
+ More information needed
 
 
31
 
32
+ ## Intended uses & limitations
 
 
 
 
 
33
 
34
+ More information needed
 
 
 
 
 
35
 
36
+ ## Training and evaluation data
37
+
38
+ More information needed
39
+
40
+ ## Training procedure
41
+
42
+ ### Training hyperparameters
43
+
44
+ The following hyperparameters were used during training:
45
+ - learning_rate: 2e-05
46
+ - train_batch_size: 16
47
+ - eval_batch_size: 32
48
+ - seed: 42
49
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
50
+ - lr_scheduler_type: linear
51
+ - num_epochs: 1
52
+
53
+ ### Framework versions
54
 
55
+ - Transformers 4.57.0
56
+ - Pytorch 2.8.0+cpu
57
+ - Datasets 4.2.0
58
+ - Tokenizers 0.22.1
SPACE_README.md CHANGED
@@ -3,7 +3,7 @@
3
  title: IMDb Sentiment Analysis
4
  emoji: ❤️
5
  colorFrom: purple
6
- colorTo: orange
7
  sdk: gradio
8
  sdk_version: 4.40.0
9
  app_file: app.py
 
3
  title: IMDb Sentiment Analysis
4
  emoji: ❤️
5
  colorFrom: purple
6
+ colorTo: pink
7
  sdk: gradio
8
  sdk_version: 4.40.0
9
  app_file: app.py
runtime.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ python-3.10
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c3e982d16820f8ce1e28616406cfd67201fb89b7f2f0367d8eb15184c4c8ec83
3
  size 5841
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb0f8e3956fa4c818ca12ff02fb511838f983b30794ed8fd8b437ce48f31f6ac
3
  size 5841