A topic tags prediction service for technical articles. The model is trained on the dataset from https://linkedinfo.co. Find the service at https://ml.linkedinfo.co/, Web API document see https://ml.linkedinfo.co/docs.
Posts relate to this project:
- Using BERT to perform Topic Tag Prediction for Technical Articles
- Multi-label classification to predict topic tags of technical articles from LinkedInfo.co
Note: these posts were performed quite some time ago, they may be now far away from good practice. So please search on the Web to find more recent ones.
Usage: service.py [OPTIONS] COMMAND [ARGS]...
Commands:
build
deploy
run
-----
Usage: service.py run [OPTIONS]
Options:
--local / --container
--help Show this message and exit.
-----
Usage: service.py build [OPTIONS]
Options:
--local / --cloud
--run
--deploy
--help Show this message and exit.
Model is trained offline and persisted. The persisted model should be placed
under ./data/models/
to be loaded by the Web service.
- [X] Language identification model for info text
- [X] Text-based info tags prediction model
- [ ] separate models for en and cn
- [ ] Graph-based info tags prediction model