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.
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.
- Install Pipenv
or simply run the following
python3 -m pip install --user pipenv
- The requirements file is in Pipefile, you can install dependency by running
pipenv install
or if it doesn't workpython -m pipenv install
- 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
- You can run python files locally on your dev machine through pipenv by
pipenv run python index.py
or activating your virtualenv bypipenv shell
- Install/Downgrade Keras to version 2.6.0 by running
pip install keras==2.6.0
- You can download base models trained on i2b2 data here. Once downloaded unzip and place content into the Models folder of MASK
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 inner_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/
Please cite the following paper:
Milosevic, N., Kalappa, G., Dadafarin, H., Azimaee, M. and Nenadic, G., 2020. MASK: A flexible framework to facilitate de-identification of clinical texts. arXiv preprint arXiv:2005.11687.
https://arxiv.org/abs/2005.11687
- We generally follow gitflow for our branching and releases
- We protect both Master and Develop branch for inadvertent code pushes
- Please make sure to enable editorconfig for your IDE
- Please try to make your pull requests and tasks as small as possible so the reviewer has easier time to understand the code.
- More documentation in your code and readme fils are always appreciated.