Owner Yann: (Experimental) Demo Google cloud application flow for covid19 data extraction from PDF's
ukehrli 5307e4a676 Delete 'notebooks/.DS_Store' | 4 years ago | |
---|---|---|
content | 4 years ago | |
notebooks | 4 years ago | |
scripts | 4 years ago | |
README.md | 4 years ago | |
requirements.txt | 4 years ago |
Medical notes and entities from TRUE patient cases publicly available on BigQuery and Datastore!
This repository contains all the code required to extract relevant information from pdf documents published by ISMIR and store raw data in a relational database and entities in a No-SQL database.
In particular, you will use Google Cloud Vision API and Translation API, before storing the information on BigQuery. Separately, you will also use specific NER models (from Scispacy) to extract (medical) domain specific entities and store them in a NoSQL db (namely Datastore) on Google Cloud Platform.
Looking for more context behind this dataset? Check out this article.
Google Cloud Architecture of the pipeline:
Quick sneak peak on the Entity dataset on Datastore:
You can replicate this pipeline directly on your local machine or on the cloud shell on GCP.
Requirements:
Modify the values to each variables in env_variables.sh file then run
cd ./covid19_ISMIR
source ./content/env_variables.sh
Set the project that you will be working on:
gcloud config set project $PROJECT_ID
Enable APIs:
gcloud services enable vision.googleapis.com
gcloud services enable translate.googleapis.com
gcloud services enable datastore.googleapis.com
gcloud services enable bigquery.googleapis.com
Install package requirements:
Make sure you have a python version >=3.6.0. Otherwise you will face some version errors Useful link
ERROR: Package 'scispacy' requires a different Python: 3.5.3 not in '>=3.6.0'
pip3 install --user -r requirements.txt
Note:
You will also need to download a Named Entity Recognition model for the second part of this pipeline. See Scispacy full selection of available models here. If you follow this installation guide, the steps will automatically download a model for you and install it.
Step 1: Download the required files to your bucket and load the required model in your local
(this step will take ~10 min)
Optional: If you have already downloaded the scispacy models, you should modify the file ./content/download_content.sh to not repeat that step
source ./content/download_content.sh
Step 2: Start the extraction of text from the pdf documents
python3 ./scripts/extraction.py
Following the extraction of text, it's time to translate it from Italian to English and curate it.
python3 ./scripts/preprocessing.py
Following the pre-processing, it's time to store the data in a more searchable format: a data warehouse - BigQuery - for the text, and a No-SQL database - Datastore - for the (UMLS) medical entities.
python3 ./scripts/storing.py True True [Model_of_your_choice]
Last but not least, this script will run a few test cases and display the results. Feel free to modify the test cases.
python3 ./scripts/retrieving.py
To get started...
Option 1
Option 2