Hugging Face's logo

CISProject
/
News-Headline-Classifier-Notebook

Safetensors
headlineclassifier
Model card Files Community
TUEN-YUE commited on
Commit
34a4bae
·
verified ·
1 Parent(s): 238cb03

Upload model

Browse files
Files changed (3) hide show
  1. README.md +3 -0
  2. config.json +3 -3
  3. model.safetensors +2 -2
README.md CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
 
2
  To test, first download the word2vec pretrained model and unzip
3
  https://drive.google.com/file/d/0B7XkCwpI5KDYNlNUTTlSS21pQmM/edit?usp=sharing
 
1
+ ---
2
+ {}
3
+ ---
4
 
5
  To test, first download the word2vec pretrained model and unzip
6
  https://drive.google.com/file/d/0B7XkCwpI5KDYNlNUTTlSS21pQmM/edit?usp=sharing
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "./exp/fox_nbc/checkpoints\\checkpoint_epoch_10",
3
  "architectures": [
4
  "CustomModel"
5
  ],
@@ -9,7 +9,7 @@
9
  "combined_dim": 128,
10
  "combined_input": 1024,
11
  "n_layers": 2,
12
- "num_classes": 1,
13
  "skip_in": [
14
  80
15
  ],
@@ -44,7 +44,7 @@
44
  "end_iter": 10,
45
  "learning_rate": 2e-05,
46
  "learning_rate_alpha": 0.05,
47
- "save_freq": 5,
48
  "val_freq": 1,
49
  "warm_up_end": 2
50
  },
 
1
  {
2
+ "_name_or_path": "./exp/fox_nbc/checkpoints\\best",
3
  "architectures": [
4
  "CustomModel"
5
  ],
 
9
  "combined_dim": 128,
10
  "combined_input": 1024,
11
  "n_layers": 2,
12
+ "num_classes": 2,
13
  "skip_in": [
14
  80
15
  ],
 
44
  "end_iter": 10,
45
  "learning_rate": 2e-05,
46
  "learning_rate_alpha": 0.05,
47
+ "save_freq": 1,
48
  "val_freq": 1,
49
  "warm_up_end": 2
50
  },
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:72ea7f79614e37d4e18699ad76b6563e7e8188a3823ff35f3d047a6b2b90c2ed
3
- size 517322024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d9eb188809c0006010d26fc47ddd73b16da1fa574d1a96797817d8d3c9867a6
3
+ size 517915672