Mirrors
/
the-algorithm
mirror of https://github.com/twitter/the-algorithm.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
the-algorithm/pushservice/src/main/python/models/heavy_ranking/README.md

1.7 KiB
Raw Permalink Blame History

Notification Heavy Ranker Model

Model Context

There are 4 major components of Twitter notifications recommendation system: 1) candidate generation 2) light ranking 3) heavy ranking & 4) quality control. This notification heavy ranker model is the core ranking model for the personalised notifications recommendation. It's a multi-task learning model to predict the probabilities that the target users will open and engage with the sent notifications.

Directory Structure

  • BUILD: this file defines python library dependencies
  • deep_norm.py: this file contains how to set up continuous training, model evaluation and model exporting for the notification heavy ranker model
  • eval.py: the main python entry file to set up the overall model evaluation pipeline
  • features.py: this file contains importing feature list and support functions for feature engineering
  • graph.py: this file defines how to build the tensorflow graph with specified model architecture, loss function and training configuration
  • model_pools.py: this file defines the available model types for the heavy ranker
  • params.py: this file defines hyper-parameters used in the notification heavy ranker
  • run_args.py: this file defines command line parameters to run model training & evaluation
  • update_warm_start_checkpoint.py: this file contains the support to modify checkpoints of the given saved heavy ranker model
  • lib/BUILD: this file defines python library dependencies for tensorflow model architecture
  • lib/layers.py: this file defines different type of convolution layers to be used in the heavy ranker model
  • lib/model.py: this file defines the module containing ClemNet, the heavy ranker model type
  • lib/params.py: this file defines parameters used in the heavy ranker model