|
--- |
|
library_name: keras |
|
language: en |
|
datasets: newsgroup |
|
tags: |
|
- multiclass-classification |
|
- newsgroup |
|
--- |
|
|
|
## Model description |
|
|
|
Text classification on the Newsgroup20 dataset using pre-trained [GloVe](https://nlp.stanford.edu/projects/glove/) word embeddings. |
|
|
|
This repo contains the model [to this Keras example using pre-trained word embeddings](https://keras.io/examples/nlp/pretrained_word_embeddings/). |
|
|
|
Full credits to : [fchollet](https://twitter.com/fchollet) |
|
|
|
Model reproduced by : [Sumedh](https://huggingface.co/sumedh) |
|
|
|
## Training and evaluation data |
|
|
|
Newsgroup20 dataset, a set of 20,000 message board messages belonging to 20 different topic categories |
|
|
|
### Training hyperparameters |
|
|
|
The following hyperparameters were used during training: |
|
|
|
| name | learning_rate | decay | rho | momentum | epsilon | centered | training_precision | |
|
|----|-------------|-----|---|--------|-------|--------|------------------| |
|
|RMSprop|0.0010000000474974513|0.0|0.8999999761581421|0.0|1e-07|False|float32| |
|
|
|
## Model Plot |
|
|
|
<details> |
|
<summary>View Model Plot</summary> |
|
|
|
![Model Image](./model.png) |
|
|
|
</details> |