COVID-19 Open Research Dataset (CORD-19) is a free resource of scholarly articles, aggregated by a coalition of leading research groups, about COVID-19 and the coronavirus family of viruses. The dataset can be found on Semantic Scholar and there is a research challenge on Kaggle.
This project builds an index over the CORD-19 dataset to assist with analysis and data discovery. A series of tasks were explored to identify relevant articles and help find answers to key scientific questions on a number of COVID-19 research topics.
The following files show the top query results for each task provided in the CORD-19 Research Challenge using this model. A highlights section is also shown for each task, which highlights the most relevant sentences from the query results.
- What is known about transmission, incubation, and environmental stability?
- What do we know about COVID-19 risk factors?
- What do we know about virus genetics, origin, and evolution?
- What do we know about vaccines and therapeutics?
- What do we know about non-pharmaceutical interventions?
- What has been published about medical care?
- What do we know about diagnostics and surveillance?
- What has been published about information sharing and inter-sectoral collaboration?
- What has been published about ethical and social science considerations?
A full overview of how to use this project can be found via this Notebook
You can use Git to clone the repository from GitHub and install it. It is recommended to do this in a Python Virtual Environment.
git clone https://github.com/neuml/cord19q.git
cd cord19q
pip install .
Python 3.5+ is supported
Download all the files in the Download CORD-19 section on Semantic Scholar. Go the directory with the files and run the following commands.
cd <download_path>
For each tar.gz file run the following, where $file is the name of the file with .tar.gz removed.
mkdir $file && tar -C $file -xvzf $file.tar.gz
Once completed, there should be a file name metadata.csv and subdirectories for each data subset with all json articles.
To build the model locally:
# Convert csv/json files to SQLite
python -m cord19q.etl <download_path>
# Can optionally use pre-trained vectors
# https://www.kaggle.com/davidmezzetti/cord19-fasttext-vectors#cord19-300d.magnitude
# Default location: ~/.cord19/vectors/cord19-300d.magnitude
python -m cord19q.vectors
# Build embeddings index
python -m cord19q.index
The model will be stored in ~/.cord19
A report file is simply a markdown file created from a list of queries. An example:
python -m cord19q.report tasks/diagnostics.txt
Once complete a file named tasks/diagnostics.md will be created.
The fastest way to run queries is to start a cord19q shell
cord19q
A prompt will come up. Queries can be typed directly into the console.
The tech stack is built on Python and creates a sentence embeddings index with FastText + BM25. Background on this method can be found in this Medium article and an existing repository using this method codequestion.
The model is a combination of the sentence embeddings index and a SQLite database with the articles. Each article is parsed into sentences and stored in SQLite along with the article metadata. FastText vectors are built over the full corpus. The sentence embeddings index only uses COVID-19 related articles, which helps produce more recent and relevant results.
Multiple entry points exist to interact with the model.
- cord19q.report - Builds a markdown report for a series of queries. For each query, the best articles are shown, top matches from those articles and a highlights section which shows the most relevant sections from the embeddings search for the query.
- cord19q.query - Runs a single query from the terminal
- cord19q.shell - Allows running multiple queries from the terminal