sumedh's picture
Update README.md
c1746b1
---
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>