Machine Learning models for Linkedinfo.co

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.

Uage

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.

Info tags prediction

Model is trained offline and persisted. The persisted model should be placed under ./data/models/ to be loaded by the Web service.

TODOs

• [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