grapefroot / quora-question-pairs

BERT for question paraphrase detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Quora-question-pairs

Solution for quora Quora question pairs challenge using pretrained BERT models

Steps to reproduce:

  1. Initialize git submodules git submodule update --init --recursive
  2. Make sure you have installed packages from requirements
  3. Download quora-pairs-dataset.zip and unzip it to ./data (create if missing)
  4. Download checkpoint weights for models from google drive model1 model2 and put them into ./models (create if missing)

Additionally, script was created to help you automate this, but in case it doesn't work for you just make above steps manually

Now, everything is ready. You may replicate the submission by running in Python

from utils import replicate
replicate(YOUR_PATH)

which will create submission csv with specified path

About

BERT for question paraphrase detection


Languages

Language:Jupyter Notebook 83.2%Language:Python 16.2%Language:Shell 0.5%