Hugging Face's logo

abhishek
/
txtclfimdb4

Text Classification
Transformers
PyTorch
TensorBoard
bert
Trained with AutoTrain
Inference Endpoints
Model card Files Metrics Community
1
abhishek commited on
Commit
29f76a9
·
1 Parent(s): af4695e

Upload folder using huggingface_hub

Browse files
Files changed (25) hide show
  1. README.md +27 -0
  2. checkpoint-196/config.json +36 -0
  3. checkpoint-196/optimizer.pt +3 -0
  4. checkpoint-196/pytorch_model.bin +3 -0
  5. checkpoint-196/rng_state_0.pth +3 -0
  6. checkpoint-196/rng_state_1.pth +3 -0
  7. checkpoint-196/scheduler.pt +3 -0
  8. checkpoint-196/trainer_state.json +41 -0
  9. checkpoint-196/training_args.bin +3 -0
  10. config.json +36 -0
  11. pytorch_model.bin +3 -0
  12. runs/Aug11_10-32-12_beast/events.out.tfevents.1691742733.beast.1746407.0 +3 -0
  13. runs/Aug11_10-33-06_beast/events.out.tfevents.1691742786.beast.1746608.0 +3 -0
  14. runs/Aug11_10-36-47_beast/events.out.tfevents.1691743009.beast.1747152.0 +3 -0
  15. runs/Aug11_11-15-04_beast/events.out.tfevents.1691745305.beast.1751000.0 +3 -0
  16. runs/Aug11_11-42-17_beast/events.out.tfevents.1691746939.beast.1753947.0 +3 -0
  17. runs/Aug11_11-46-02_beast/events.out.tfevents.1691747163.beast.1754528.0 +3 -0
  18. runs/Aug11_11-46-34_beast/events.out.tfevents.1691747195.beast.1754668.0 +3 -0
  19. runs/Aug11_11-49-18_beast/events.out.tfevents.1691747358.beast.1755130.0 +3 -0
  20. runs/Aug11_12-14-28_beast/events.out.tfevents.1691748869.beast.1757815.0 +3 -0
  21. runs/Aug11_12-16-04_beast/events.out.tfevents.1691748965.beast.1758195.0 +3 -0
  22. runs/Aug11_12-50-38_beast/events.out.tfevents.1691751039.beast.1762496.0 +3 -0
  23. runs/Aug11_12-50-38_beast/events.out.tfevents.1691751130.beast.1762496.1 +3 -0
  24. training_args.bin +3 -0
  25. training_params.json +29 -0
README.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - text-classification
6
+ widget:
7
+ - text: "I love AutoTrain"
8
+ datasets:
9
+ - imdb
10
+ ---
11
+
12
+ # Model Trained Using AutoTrain
13
+
14
+ - Problem type: Text Classification
15
+
16
+ ## Validation Metrics
17
+ loss: 0.2895243167877197
18
+
19
+ f1: 0.8814423039802339
20
+
21
+ precision: 0.85174960829665
22
+
23
+ recall: 0.91328
24
+
25
+ auc: 0.9512297792000001
26
+
27
+ accuracy: 0.87716
checkpoint-196/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-uncased",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "BertForSequenceClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.1,
8
+ "classifier_dropout": null,
9
+ "gradient_checkpointing": false,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "id2label": {
14
+ "0": 0,
15
+ "1": 1
16
+ },
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 3072,
19
+ "label2id": {
20
+ "0": 0,
21
+ "1": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 512,
25
+ "model_type": "bert",
26
+ "num_attention_heads": 12,
27
+ "num_hidden_layers": 12,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "problem_type": "single_label_classification",
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.32.0.dev0",
33
+ "type_vocab_size": 2,
34
+ "use_cache": true,
35
+ "vocab_size": 30522
36
+ }
checkpoint-196/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bdc8cdde6354bbc81e5893ff1c95a54f761ec1c77e4ebcfbc48eb6f6a1f53ec7
3
+ size 876037893
checkpoint-196/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82b4abb9ed3963b09f07f4f4653ee9a47bb5fa2c1e6b9a3352614e9f5e492c6c
3
+ size 438003505
checkpoint-196/rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f07bd0489570e4ff705507c352b86c94b1a4b5b2e4c18d949e98ae7a84916f0
3
+ size 15607
checkpoint-196/rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e9ad1cf33d456b78ec00e692f2f4a9039d52e8bbc791ce7e4e83e077d7f7146
3
+ size 15607
checkpoint-196/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:318cd6d635fce44872177e69a182b30dcec98234ea257198a9d6c22c6978eb87
3
+ size 627
checkpoint-196/trainer_state.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.2895243167877197,
3
+ "best_model_checkpoint": "output/checkpoint-196",
4
+ "epoch": 1.0,
5
+ "global_step": 196,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.4,
12
+ "learning_rate": 2.0113636363636362e-05,
13
+ "loss": 0.4584,
14
+ "step": 78
15
+ },
16
+ {
17
+ "epoch": 0.8,
18
+ "learning_rate": 6.818181818181818e-06,
19
+ "loss": 0.3111,
20
+ "step": 156
21
+ },
22
+ {
23
+ "epoch": 1.0,
24
+ "eval_accuracy": 0.87716,
25
+ "eval_auc": 0.9512297792000001,
26
+ "eval_f1": 0.8814423039802339,
27
+ "eval_loss": 0.2895243167877197,
28
+ "eval_precision": 0.85174960829665,
29
+ "eval_recall": 0.91328,
30
+ "eval_runtime": 18.4052,
31
+ "eval_samples_per_second": 1358.312,
32
+ "eval_steps_per_second": 5.325,
33
+ "step": 196
34
+ }
35
+ ],
36
+ "max_steps": 196,
37
+ "num_train_epochs": 1,
38
+ "total_flos": 1650232534237184.0,
39
+ "trial_name": null,
40
+ "trial_params": null
41
+ }
checkpoint-196/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ef40698ce2bb20bea547ddc158158a30d219c6e97bc79a98d08defc733b7b4
3
+ size 3963
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-uncased",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "BertForSequenceClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.1,
8
+ "classifier_dropout": null,
9
+ "gradient_checkpointing": false,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "id2label": {
14
+ "0": 0,
15
+ "1": 1
16
+ },
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 3072,
19
+ "label2id": {
20
+ "0": 0,
21
+ "1": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 512,
25
+ "model_type": "bert",
26
+ "num_attention_heads": 12,
27
+ "num_hidden_layers": 12,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "problem_type": "single_label_classification",
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.32.0.dev0",
33
+ "type_vocab_size": 2,
34
+ "use_cache": true,
35
+ "vocab_size": 30522
36
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82b4abb9ed3963b09f07f4f4653ee9a47bb5fa2c1e6b9a3352614e9f5e492c6c
3
+ size 438003505
runs/Aug11_10-32-12_beast/events.out.tfevents.1691742733.beast.1746407.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e0a81090d6a6ac29ba2edc627805d85fced658bbf1b93d97d36d787dc3d3bff
3
+ size 4135
runs/Aug11_10-33-06_beast/events.out.tfevents.1691742786.beast.1746608.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c623ecfb5684fc3378fd28df840135181b4331a7a91bdf6199f9f83625b3b06
3
+ size 4135
runs/Aug11_10-36-47_beast/events.out.tfevents.1691743009.beast.1747152.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09ff46130d623e1862bafbd62b684de3a968c2c7b03bf620f3ae929c8ea0ea76
3
+ size 4804
runs/Aug11_11-15-04_beast/events.out.tfevents.1691745305.beast.1751000.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d451ccb2a6ba36ef985be8c1d409390fc9576d82abc9847a2d3b9750c802ce8
3
+ size 4804
runs/Aug11_11-42-17_beast/events.out.tfevents.1691746939.beast.1753947.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d68f9b3ab08dee2b8d9a4dd4dca4d3d8b3d5a07e49566e8069f18be3d356abd
3
+ size 4136
runs/Aug11_11-46-02_beast/events.out.tfevents.1691747163.beast.1754528.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df4fd997666c1b14d2e2b467ae0a6bdeb7f1bda3acfb6e225c56039859de2209
3
+ size 88
runs/Aug11_11-46-34_beast/events.out.tfevents.1691747195.beast.1754668.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97833253f5fe42e6637e00cf0243d1c287aaf95787e644f8216bfe215ad836a3
3
+ size 88
runs/Aug11_11-49-18_beast/events.out.tfevents.1691747358.beast.1755130.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c379970ef4f7b8ecafe51f8836215c405c88f36224c01bab68f40a4aefc97472
3
+ size 4804
runs/Aug11_12-14-28_beast/events.out.tfevents.1691748869.beast.1757815.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e05e15035530b93d1b8dbe4e8f1a548a66715432b1b346bb23e38e943dab3ac
3
+ size 4802
runs/Aug11_12-16-04_beast/events.out.tfevents.1691748965.beast.1758195.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a768ef9c3925bc2783a64abdd535cb6a8cf45db7f1a77e093c71e4f1ff541ca4
3
+ size 4802
runs/Aug11_12-50-38_beast/events.out.tfevents.1691751039.beast.1762496.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cb0cfe56b80121df7e4633bdaa0e56571ac4a2cd9f588cd41a0a90af659bbbd
3
+ size 5325
runs/Aug11_12-50-38_beast/events.out.tfevents.1691751130.beast.1762496.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bec68b85fc646534f703301e5890fe48cc6940eb1a1b4d03540ceb27e36c5377
3
+ size 607
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ef40698ce2bb20bea547ddc158158a30d219c6e97bc79a98d08defc733b7b4
3
+ size 3963
training_params.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "imdb",
3
+ "model_name": "bert-base-uncased",
4
+ "lr": 3e-05,
5
+ "epochs": 1,
6
+ "max_seq_length": 128,
7
+ "batch_size": 64,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": "test",
17
+ "text_column": "text",
18
+ "target_column": "label",
19
+ "logging_steps": -1,
20
+ "project_name": "output",
21
+ "auto_find_batch_size": false,
22
+ "fp16": true,
23
+ "save_total_limit": 1,
24
+ "save_strategy": "epoch",
25
+ "token": null,
26
+ "push_to_hub": true,
27
+ "repo_id": "abhishek/txtclfimdb4",
28
+ "evaluation_strategy": "epoch"
29
+ }