Hugging Face's logo

doanvinhlong18
/
lstm-news-classifier

Text Classification
Keras
English
news
lstm
tensorflow
Model card Files
xet
 Community
doanvinhlong18 commited on
Commit
4b896a0
·
verified ·
1 Parent(s): bc889ae

Add fine-tuned Bidirectional LSTM news classifier

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. keras_tokenizer.pkl +1 -1
  3. lstm_model.keras +1 -1
README.md CHANGED
@@ -29,8 +29,8 @@ on the [HuffPost News Category Dataset](https://www.kaggle.com/datasets/rmisra/n
29
  | Vocab size | 60,000 |
30
  | Max token length | 256 |
31
  | Embedding dim | 128 |
32
- | Test Accuracy | 0.6903 |
33
- | Test F1 (macro) | 0.6154 |
34
 
35
  ## Labels
36
 
 
29
  | Vocab size | 60,000 |
30
  | Max token length | 256 |
31
  | Embedding dim | 128 |
32
+ | Test Accuracy | 0.6909 |
33
+ | Test F1 (macro) | 0.6191 |
34
 
35
  ## Labels
36
 
keras_tokenizer.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6803beeb8ce72be4c452bd28c4b837aef7ce7d9a0039b1623ed3547818eaab47
3
  size 3147868
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19ab7cf1a2cfa1fa78be8ebf8605d9d9d4d4cd9606220d1d426e4c8ee6496c25
3
  size 3147868
lstm_model.keras CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f3d27cd4e193beb674e9f100b71c7df11364611fe862d2ffe62a2d08405f8f10
3
  size 97575439
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3bc0a91ea92c0c302a331bf8f345baf771da9ca03ba3b3de1e54829bd0ed1c3b
3
  size 97575439