[735bb5]: / src / ml_models / README.md

Download this file

14 lines (8 with data), 666 Bytes

Machine Learning Methods

This directory contains the code for the Random Forest and BiLSTM methods.

Random Forest

The Random Forest method is based on scikit-learn's implementation.

BiLSTM

The BiLSTM method has been implemented using the PyTorch library using some already-implemented modules from the Joey NMT library.

Clinical BERT & Paired Clinical BERT

We use the Transformers BERT implementation for the BERT-based methods.