Skip to content

Commit efede91

Browse files
committed
Add images
1 parent 4ebbe76 commit efede91

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@ This raw dataset can be found in ``data/tweets.csv``.
1616

1717
## 2. The model
1818
As it was already commented, the aim of this repository is to provide a base line model for text classfication. In this sense, the model is based on a two-stacked LSTM layers followed by two linear layers. The dataset is preprocessed through a tokens-based technique, then tokens are associated to an embedding layer. The following image describes the pipeline of the model.
19+
<p align="center">
20+
<img src='img/model_architecture.png'>
21+
</p>
1922

2023
## 3. Install dependencies
2124
Working on

img/model_architecture.png

262 KB
Loading

0 commit comments

Comments
 (0)