Hugging Face's logo

CISProject
/
News-Headline-Classifier-Notebook

Safetensors
headlineclassifier
Model card Files Community
TUEN-YUE commited on
Commit
da99c05
·
verified ·
1 Parent(s): 4a8afb6

Upload model

Browse files
Files changed (2) hide show
  1. config.json +64 -64
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,64 +1,64 @@
1
- {
2
- "_name_or_path": "/content/drive/My Drive/CIS5190/exp/fox_nbc/checkpoints/checkpoint_epoch_10",
3
- "architectures": [
4
- "CustomModel"
5
- ],
6
- "base_exp_dir": "/content/drive/My Drive/CIS5190/exp/fox_nbc/",
7
- "model": {
8
- "cls": {
9
- "combined_dim": 128,
10
- "combined_input": 384,
11
- "n_layers": 4,
12
- "num_classes": 1,
13
- "skip_in": [
14
- 8
15
- ],
16
- "weight_norm": true
17
- },
18
- "freq": {
19
- "n_layers": 4,
20
- "skip_in": [
21
- 10
22
- ],
23
- "tfidf_hidden_dim": 512,
24
- "tfidf_input_dim": 7725,
25
- "tfidf_output_dim": 128,
26
- "weight_norm": true
27
- },
28
- "pos": {
29
- "hidden_dim": 256,
30
- "input_dim": 512,
31
- "n_layers": 4,
32
- "output_dim": 128,
33
- "skip_in": [
34
- 10
35
- ],
36
- "weight_norm": true
37
- },
38
- "seq": {
39
- "freeze": true,
40
- "hidden_dim": 256,
41
- "input_dim": 512,
42
- "lstm_in": 768,
43
- "n_layers": 4,
44
- "output_dim": 128,
45
- "skip_in": [
46
- 10
47
- ],
48
- "weight_norm": true
49
- }
50
- },
51
- "model_type": "classifier",
52
- "torch_dtype": "float32",
53
- "train": {
54
- "anneal_end": 5,
55
- "batch_size": 32,
56
- "end_iter": 10,
57
- "learning_rate": 0.0005,
58
- "learning_rate_alpha": 0.05,
59
- "save_freq": 5,
60
- "val_freq": 2,
61
- "warm_up_end": 2
62
- },
63
- "transformers_version": "4.46.3"
64
- }
 
1
+ {
2
+ "_name_or_path": "/content/drive/My Drive/CIS5190/exp/fox_nbc/checkpoints\\checkpoint_epoch_10",
3
+ "architectures": [
4
+ "CustomModel"
5
+ ],
6
+ "base_exp_dir": "/content/drive/My Drive/CIS5190/exp/fox_nbc/",
7
+ "model": {
8
+ "cls": {
9
+ "combined_dim": 128,
10
+ "combined_input": 256,
11
+ "n_layers": 4,
12
+ "num_classes": 1,
13
+ "skip_in": [
14
+ 8
15
+ ],
16
+ "weight_norm": true
17
+ },
18
+ "freq": {
19
+ "n_layers": 4,
20
+ "skip_in": [
21
+ 10
22
+ ],
23
+ "tfidf_hidden_dim": 512,
24
+ "tfidf_input_dim": 7725,
25
+ "tfidf_output_dim": 128,
26
+ "weight_norm": true
27
+ },
28
+ "pos": {
29
+ "hidden_dim": 256,
30
+ "input_dim": 512,
31
+ "n_layers": 4,
32
+ "output_dim": 128,
33
+ "skip_in": [
34
+ 10
35
+ ],
36
+ "weight_norm": true
37
+ },
38
+ "seq": {
39
+ "freeze": true,
40
+ "hidden_dim": 256,
41
+ "input_dim": 512,
42
+ "lstm_in": 768,
43
+ "n_layers": 4,
44
+ "output_dim": 128,
45
+ "skip_in": [
46
+ 10
47
+ ],
48
+ "weight_norm": true
49
+ }
50
+ },
51
+ "model_type": "headlineclassifier",
52
+ "torch_dtype": "float32",
53
+ "train": {
54
+ "anneal_end": 5,
55
+ "batch_size": 32,
56
+ "end_iter": 10,
57
+ "learning_rate": 0.0005,
58
+ "learning_rate_alpha": 0.05,
59
+ "save_freq": 5,
60
+ "val_freq": 2,
61
+ "warm_up_end": 2
62
+ },
63
+ "transformers_version": "4.47.0"
64
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:313dae52124ff7952ed4d0013f61866d67ac1f0f8533173473e53e8b061a2c76
3
- size 471001816
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fcae66beb513dc539d22658af4be87f9405e9ef6b0be594b2c9ca7fd4eb9eee
3
+ size 469488864