Skip to content

The LSTM-DSSM model refering to the paper: Deep Sentence Embedding Using Long Short-Term Memory Networks: Analysis and Application to Information Retrieval

Notifications You must be signed in to change notification settings

peng3307165/dssm-lstm

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

The LSTM-DSSM model refering to the paper: Deep Sentence Embedding Using Long Short-Term Memory Networks: Analysis and Application to Information Retrieval.

papaer available at: https://arxiv.org/abs/1502.06922

data: https://www.kaggle.com/c/quora-question-pairs/data, only use train.csv.

word vector: glove.6B(300d)

rename train.csv to all.csv, rename the word vector file to vector.txt, and put them under data dir

then run pre_procession.py to generate queries.txt, docs.txt, test_queries.txt, test_docs.txt, test_ground_truths.txt, validate_queries.txt, validate_docs.txt and validate_ground_truths.txt

pre_procession is written under py3 because I'm too lazy that I only get nltk installed under my py3 environment, I don't know if it works under py2. the other part of the program is written in py2

in test we use log loss according to this page: https://www.kaggle.com/wiki/LogLoss

loss decreased on train set but keep increasing on validate set, haven't figure out why. Maybe the data set is too small

About

The LSTM-DSSM model refering to the paper: Deep Sentence Embedding Using Long Short-Term Memory Networks: Analysis and Application to Information Retrieval

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%