Diff of /src/ml_models/README.md [000000] .. [735bb5]

Switch to unified view

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.