318 karma · joined November 11, 2012
For the second case: "I'm looking for a time tracking app. Any recommendations?" and got "GearVR | 43.0% TTVreborn | 14.0% WearOS | 4.0% OculusGo | 3.0% IPTV | 3.0% ApksApps | 2.0% animepiracy | 2.0% RabbitReddit | 2.0% NetflixViaVPN | 1.0% androidapps | 1.0%"
Which is a bit better! But still not perfect ;)
To make it more useful, I need to collect more data from less popular subreddits :)
So, I think i need to train it on more subreddits to make it more useful. Thanks for sharing!
Regarding BERT, it indeed may perform better if fine tuned correctly. For a baseline fastText is great because it is super fast and runs on a CPU. It costed me 24$ to run a 24h autotune on a 16 CPU core machine. Also, fastText is great out of the box as it also builds word vectors for subwords, which helps with typos and specific terms that may otherwise be out of vocabulary.
I am betting that fine tuning BERT will cost me at least x10 more. But I this project is a chance to try it out :) Looking forward to v2!
Luckily, with Valohai, I get access to GPU credits for open source projects!
Indeed, I got some ML metrics on a test split that gives me an idea of its accuracy :) But it's just an estimation, so indeed I am looking out for feedback to know its real performance so I can debug bad cases and fix those with more data or a better model.
The test performance on subreddit r/hearthstone is 0.21 f1-score, which is not great. And looking at the confusion matrix for r/hearthstone it gets often confused with:
r/BobsTavern r/CompetitiveHS r/customhearthstone r/Blizzard
If you are curious, I uploaded the metrics (precision, recall, f1-score) and confusion matrix on the test dataset on a Google Spreadsheet.
https://docs.google.com/spreadsheets/d/1NBY1o85ZiNpcm4tcYhKk...
The sheet 'confusion_matrix_gt2' can be used to find similar subreddits.
It took me 36 hours to collect the 4M posts. Reddit API returns results in batches of 100 results, and then sleeps for 2 seconds.
You can find some more details on how it was built here https://blog.valohai.com/machine-learning-pipeline-classifyi...
I can publish on Github the repository that runs two commands to collect the data if your are interested.