Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,19 @@
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
| 3 |
---
|
| 4 |
+
|
| 5 |
+
# Pat2Vec
|
| 6 |
+
Fro a description of the framework and model, see our publication: <https://preprints.jmir.org/preprint/40755/>
|
| 7 |
+
|
| 8 |
+
It is trained using the amazing gensim package version 4 and parameters were optimized with Bayesian optimization (using another amazing package, optuna).
|
| 9 |
+
|
| 10 |
+
Unfortunately, this gensim model cannot be loaded directly. You have to clone the repository or download all the files and run the following to use it.
|
| 11 |
+
|
| 12 |
+
## quick start to use the model in Python:
|
| 13 |
+
```
|
| 14 |
+
from gensim.models.doc2vec import Doc2Vec
|
| 15 |
+
|
| 16 |
+
pat2vec_model = Doc2Vec.load('pat2vec_dim10.model')
|
| 17 |
+
|
| 18 |
+
pat2vec_model.infer_vector(["M54.1", "J06.9", "I10.90", "R51"])
|
| 19 |
+
```
|