dieineb commited on
Commit
58fabde
1 Parent(s): fe5e83e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -15
README.md CHANGED
@@ -4,26 +4,27 @@ tags:
4
  - translation
5
  license: apache-2.0
6
  ---
7
- # Model Description
8
 
9
- The GRU-eng-por model is used to translate English to Portuguese automatically.
10
 
11
- The model was trained with a traduction dataset from English to Portuguese.
12
 
13
- ## Details
14
- - Size: 42,554,912 parameters
15
- - Dataset: [`English-to-Portuguese`](https://www.kaggle.com/datasets/nageshsingh/englishportuguese-translation)
16
- - Languages: English, Portuguese
17
- - Number of Epochs: 15
18
- - Batch size: 32
19
- - Optimizer: rmsprop
20
- - Learning Rate: 0.001
21
- - Hardware: Tesla T4
22
- - This repository has the source [code used](https://github.com/Nkluge-correa/teeny-tiny_castle/blob/master/ML%20Intro%20Course/16_sequence_to_sequence.ipynb) to train this model.
23
 
24
- ## Usage
 
 
 
 
 
 
 
 
25
 
26
- ```
 
 
27
  !pip install huggingface_hub["tensorflow"] -q
28
 
29
  from huggingface_hub import from_pretrained_keras
@@ -154,6 +155,7 @@ Portuguese translation:
154
  --------------------------------------------------
155
  ```
156
 
 
157
  # Cite as 🤗
158
  ```
159
  @misc{teenytinycastle,
 
4
  - translation
5
  license: apache-2.0
6
  ---
7
+ # GRU-eng-por
8
 
9
+ ## Model Overview
10
 
11
+ The GRU-eng-por model is a Recurrent Neural Network (RNN) for machine translation tasks (English-Portuguese).
12
 
13
+ ### Details
 
 
 
 
 
 
 
 
 
14
 
15
+ - **Size:** 42,554,912 parameters
16
+ - **Model type:** Recurrent neural network
17
+ - **Optimizer**: `rmsprop`
18
+ - **Number of Epochs:** 15
19
+ - **Dimensionality of the embedding layer** = 256
20
+ - **dimensionality of the feed-forward network** = 1024
21
+ - **Hardware:** Tesla T4
22
+ - **Emissions:** Not measured
23
+ - **Total Energy Consumption:** Not measured
24
 
25
+ ### How to Use
26
+
27
+ ```python
28
  !pip install huggingface_hub["tensorflow"] -q
29
 
30
  from huggingface_hub import from_pretrained_keras
 
155
  --------------------------------------------------
156
  ```
157
 
158
+
159
  # Cite as 🤗
160
  ```
161
  @misc{teenytinycastle,