|
a |
|
b/src/ml_models/README.md |
|
|
1 |
# Machine Learning Methods |
|
|
2 |
|
|
|
3 |
This directory contains the code for the Random Forest and BiLSTM methods. |
|
|
4 |
|
|
|
5 |
## Random Forest |
|
|
6 |
The Random Forest method is based on [scikit-learn's implementation](https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html). |
|
|
7 |
|
|
|
8 |
|
|
|
9 |
## BiLSTM |
|
|
10 |
The BiLSTM method has been implemented using the [PyTorch](https://pytorch.org/) library using some already-implemented modules from the [Joey NMT](https://github.com/joeynmt/joeynmt) library. |
|
|
11 |
|
|
|
12 |
|
|
|
13 |
## Clinical BERT & Paired Clinical BERT |
|
|
14 |
We use the [Transformers BERT](https://huggingface.co/docs/transformers/model_doc/bert) implementation for the BERT-based methods. |