Skip to content
forked from hwwang55/DKN

A tensorflow implementation of DKN (Deep Knowledge-aware Network for News Recommendation)

Notifications You must be signed in to change notification settings

NinedayWang/DKN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 

Repository files navigation

DKN

This repository is the implementation of DKN (arXiv):

DKN: Deep Knowledge-Aware Network for News Recommendation
Hongwei Wang, Fuzheng Zhang, Xing Xie, Minyi Guo
The Web Conference 2018 (WWW 2018)

DKN is a deep knowledge-aware network that takes advantage of knowledge graph representation in news recommendation. The main components in DKN is a KCNN module and an attention module:

  • The KCNN module is to learn from semantic-level and knowledge-level representations of news jointly. The multiple channels and alignment of words and entities enable KCNN to combine information from heterogeneous sources.
  • The attention module is to model the different impacts of a user’s diverse historical interests on current candidate news.

Files in the folder

  • data/
    • kg/
      • Fast-TransX: an efficient implementation of TransE and its extended models for Knowledge Graph Embedding (from https://github.com/thunlp/Fast-TransX);
      • kg.txt: knowledge graph file;
      • kg_preprocess.py: pre-process the knowledge graph and output knowledge embedding files for DKN;
      • prepare_data_for_transx.py: generate the required input files for Fast-TransX;
    • news/
      • news_preprocess.py: pre-process the news dataset;
      • raw_test.txt: raw test data file;
      • raw_train.txt: raw train data file;
  • src/: implementations of DKN.

Note: Due to the pricacy policies of Bing News and file size limits on Github, the released raw dataset and the knowledge graph in this repository is only a small sample of the original ones reported in the paper.

Format of input files

  • raw_train.txt and raw_test.txt:
    user_id[TAB]news_title[TAB]label[TAB]entity_info
    for each line, where news_title is a list of words w1 w2 ... wn, and entity_info is a list of pairs of entity id and entity name: entity_id_1:entity_name;entity_id_2:entity_name...
  • kg.txt:
    head[TAB]relation[TAB]tail
    for each line, where head and tail are entity ids and relation is the relation id.

Required packages

The code has been tested running under Python 3.6.5, with the following packages installed (along with their dependencies):

  • tensorflow-gpu == 1.4.0
  • numpy == 1.14.5
  • sklearn == 0.19.1
  • pandas == 0.23.0
  • gensim == 3.5.0

Running the code

$ cd data/news
$ python news_preprocess.py
$ cd ../kg
$ python prepare_data_for_transx.py
$ cd Fast-TransX/transE/ (note: you can also choose other KGE methods)
$ g++ transE.cpp -o transE -pthread -O3 -march=native
$ ./transE
$ cd ../..
$ python kg_preprocess.py
$ cd ../../src
$ python main.py (note: use -h to check optional arguments)

About

A tensorflow implementation of DKN (Deep Knowledge-aware Network for News Recommendation)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 68.2%
  • Python 31.8%