Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ The rules for each genre are defined below:
|
|
37 |
|
38 |
## Data
|
39 |
|
40 |
-
A
|
41 |
|
42 |
For more detail, refer to the _Acknowledgments_ section
|
43 |
|
@@ -75,7 +75,7 @@ poem = '''Người đi theo gió đuổi mây
|
|
75 |
|
76 |
classifier.predict(poem)
|
77 |
|
78 |
-
#>> [{'label': 'luc bat', 'confidence': 0.9999017715454102, 'poem_score': 0.75}]
|
79 |
```
|
80 |
|
81 |
## Model
|
@@ -84,6 +84,6 @@ The model's weights are published at Huggingface [Anshler/vietnamese-poem-classi
|
|
84 |
|
85 |
## Acknowledgments
|
86 |
|
87 |
-
_This project was inspired by the evaluation method from ```fsoft-ailab```
|
88 |
|
89 |
-
_Dataset also taken from their repo_
|
|
|
37 |
|
38 |
## Data
|
39 |
|
40 |
+
A collection of 171188 Vietnamese poems with different genres: luc-bat, 5-chu, 7-chu, 8-chu, 4-chu. Download [here](https://github.com/fsoft-ailab/Poem-Generator/raw/master/dataset/poems_dataset.zip)
|
41 |
|
42 |
For more detail, refer to the _Acknowledgments_ section
|
43 |
|
|
|
75 |
|
76 |
classifier.predict(poem)
|
77 |
|
78 |
+
#>> [{'label': 'luc bat', 'confidence': 0.9999017715454102, 'poem_score': 0.75, 'l_score': 1.0, 't_score': 1.0, 'r_score': 0.5833333333333333}]
|
79 |
```
|
80 |
|
81 |
## Model
|
|
|
84 |
|
85 |
## Acknowledgments
|
86 |
|
87 |
+
_This project was inspired by the evaluation method from ```fsoft-ailab```'s_ [SP-GPT2 Poem-Generator](https://github.com/fsoft-ailab/Poem-Generator)
|
88 |
|
89 |
+
_Dataset also taken from their repo_
|