File size: 1,111 Bytes
67940a2
c1746b1
ad46870
 
67940a2
ad46870
 
67940a2
 
 
 
c1746b1
67940a2
c1746b1
67940a2
c1746b1
67940a2
c1746b1
67940a2
c1746b1
67940a2
c1746b1
67940a2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
---
datasets: newsgroup
language: en
library_name: tf-keras
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>