Loading project…
Loading project…
I cleaned 2,838 posts, built a bag-of-words matrix with 7,842 terms, trained a Bernoulli naive Bayes classifier and tried it on two sentences I typed.
Original tutorial by Aman Kharwal: Stress Detection (opens in a new tab)
| Output | Value | Cell # |
|---|---|---|
| Posts vectorised | 2,838 | |
| Vocabulary size | 7,842 |
Cell # counts every cell from the top of the notebook, Markdown included, starting at 1.
No accuracy was ever printed, so there is nothing to judge the model by.
Labelled Reddit posts from mental-health and relationship communities, re-uploaded from the Dreaddit research dataset.
The posts describe people's mental health. None of the text is shown on this site.