/MASK_public

Primary LanguagePythonApache License 2.0Apache-2.0

Build Status PWC

Mask Project

Mask project is a collaborative project between Institute for Clinical Evaluative Sciences, University of Manchester and Evenset Custom Medical Software Development. The purpose of this project is masking identifiable information from health related documents.

Installation

Using a virtual environment to install project dependency is highly recommend. We're using Pipenv for python package management to ensure more stable dependency management. Read more. Python 3.7 is recommended to run this package.

  1. Install Pipenv or simply run the following python3 -m pip install --user pipenv
  2. The requirements file is in Pipefile, you can install dependency by running pipenv install
  3. Sometimes lock file subprocess hangs, you can avoid it by running your installation with something like this PIP_NO_CACHE_DIR=off pipenv install keras==2.2.4
  4. You can run python files locally on your dev machine through pipenv by pipenv run python index.py or activating your virtualenv by pipenv shell

Running

There are two main files that are used to run Mask:

  • train_framework.py - This file is used to train named entity recognizers that will be used for masking. It contains a set if command line parameters and it uses the classes in ner_plugins folder. For each model, it is advisable to change a name of the model file in the appropriate file/class in ner_plugins folder. This is the example of how to run this script: python train_framework.py --source-type i2b2 --source_location "<relative location to training files in i2b2 format>" --algorithm NER_Name_of_the_algo --do_test yes --save_model yes --epochs 5

More concretly:

python train_framework.py --source-type i2b2 --source_location "dataset/i2b2/" --algorithm NER_CRF --do_test yes --save_model yes --epochs 5

  • mask_framework.py - This file is used to run NER and masking. As input it uses a set of text files and outputs a set of text fules. Input and output paths, as well as masking and NER algorithms are defined in configuration.cnf file.

Training of NER algorithms is at the moment supported only if in i2b2 format (other format need to be converted to this format). i2b2 2014, which have been used in development of this tool can be requested at the following location: https://portal.dbmi.hms.harvard.edu/projects/n2c2-nlp/

Contribution

Git Branching

  1. We generally follow gitflow for our branching and releases
  2. We protect both Master and Develop branch for inadvertent code pushes

Coding guideline

  1. Please make sure to enable editorconfig for your IDE
  2. Please try to make your pull requests and tasks as small as possible so the reviewer has easier time to understand the code.
  3. More documentation in your code and readme fils are always appreciated.

Contributors

  1. Nikola Milosevic
  2. Hesam Dadafarin