nnnyt / MIND

News classification & recommendation in Keras

MIND

Dataset: MIND: MIcrosoft News Dataset

Part1 News Classification

Implemented:

• HAN: Hierarchical Attention Networks for Document Classification

  Classify news into categories (17):

  $ python part1/HAN/han.py

  Classify news into subcategories (264):

  $ python part1/HAN/han_sub.py

• News classification based on NAML

  Classify news into categories:

  $ python part1/NAML/NAML.py

  Classify news into subcategories:

  $ python part1/NAML/NAML_sub.py

Part2 News Recommendation

Implemented:

• NAML: Neural News Recommendation with Attentive Multi-view Learning

  $ python part2/NAML/NAML.py

• NRMS: Neural News Recommendation with Multi-Head Self-Attention

  $ python part2/NRMS/NRMS.py