leonardo
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,31 @@
|
|
1 |
-
---
|
2 |
-
license: cc-by-3.0
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: cc-by-3.0
|
3 |
+
datasets:
|
4 |
+
- stanfordnlp/imdb
|
5 |
+
language:
|
6 |
+
- en
|
7 |
+
metrics:
|
8 |
+
- accuracy
|
9 |
+
---
|
10 |
+
|
11 |
+
model structure:
|
12 |
+
|
13 |
+
Input Text
|
14 |
+
|
|
15 |
+
Custom Tokenizer (Text -> Tokens)
|
16 |
+
|
|
17 |
+
Padding (Equal Length Sequences)
|
18 |
+
|
|
19 |
+
Embedding Layer (Tokens -> Dense Vectors)
|
20 |
+
|
|
21 |
+
LSTM Layer (Capture Sequential Patterns)
|
22 |
+
|
|
23 |
+
Attention Mechanism (Focus on Important Tokens)
|
24 |
+
|
|
25 |
+
Dense Layer (Fully Connected)
|
26 |
+
|
|
27 |
+
Softmax Activation (Output: Positive or Negative Sentiment)
|
28 |
+
|
29 |
+
|
30 |
+
it archieved 95.71% accuracy on the dataset provided
|
31 |
+
|