Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,7 @@ inference: false
|
|
8 |
|
9 |
<!-- Provide a quick summary of what the model is/does. -->
|
10 |
|
11 |
-
This model is intended to predict emotions (valence, arousal) in written stories. For all details see [the paper (TODO)](#) and [the accompanying github repo
|
12 |
|
13 |
|
14 |
|
@@ -37,14 +37,14 @@ are meant to be the predictions for the sentence preceding it. All other tokens'
|
|
37 |
|
38 |
![image](tales_vertical.png)
|
39 |
|
40 |
-
The [accompanying repo](
|
41 |
|
42 |
### Model Sources
|
43 |
|
44 |
<!-- Provide the basic links for the model. -->
|
45 |
|
46 |
-
- **Repository:** [Github](
|
47 |
-
- **Paper:** [ArXiv](
|
48 |
|
49 |
## Uses
|
50 |
|
@@ -53,7 +53,7 @@ This model is intended to predict emotions (valence, arousal) in written stories
|
|
53 |
Please note that the model is not production-ready and provided here for demonstration purposes only.
|
54 |
For details on the datasets used, please refer to the [paper (TODO)](#).
|
55 |
|
56 |
-
In the [github repository](
|
57 |
|
58 |
``
|
59 |
python3 predict.py --input_csv input_file.csv --output_csv output_file.csv --checkpoint_dir chrlukas/stories-emotion-c4 --window_size 4 --batch_size 4
|
|
|
8 |
|
9 |
<!-- Provide a quick summary of what the model is/does. -->
|
10 |
|
11 |
+
This model is intended to predict emotions (valence, arousal) in written stories. For all details see [the paper (TODO)](#) and [the accompanying github repo](https://github.com/lc0197/emotional_trajectories_stories).
|
12 |
|
13 |
|
14 |
|
|
|
37 |
|
38 |
![image](tales_vertical.png)
|
39 |
|
40 |
+
The [accompanying repo](https://github.com/lc0197/emotional_trajectories_stories) provides a convenient script to use the model for prediction.
|
41 |
|
42 |
### Model Sources
|
43 |
|
44 |
<!-- Provide the basic links for the model. -->
|
45 |
|
46 |
+
- **Repository:** [Github](https://github.com/lc0197/emotional_trajectories_stories)
|
47 |
+
- **Paper:** [ArXiv](#)
|
48 |
|
49 |
## Uses
|
50 |
|
|
|
53 |
Please note that the model is not production-ready and provided here for demonstration purposes only.
|
54 |
For details on the datasets used, please refer to the [paper (TODO)](#).
|
55 |
|
56 |
+
In the [github repository](https://github.com/lc0197/emotional_trajectories_stories), a convenient script to predict V/A in existing texts is provided. Example call:
|
57 |
|
58 |
``
|
59 |
python3 predict.py --input_csv input_file.csv --output_csv output_file.csv --checkpoint_dir chrlukas/stories-emotion-c4 --window_size 4 --batch_size 4
|