| --- |
| license: cc-by-4.0 |
| --- |
| |
| # Pat2Vec |
| Fro a description of the framework and model, see our publication: <https://preprints.jmir.org/preprint/40755/> |
|
|
| It is trained using the amazing gensim package version 4 and parameters were optimized with Bayesian optimization (using another amazing package, optuna). |
|
|
| 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. |
|
|
| ## quick start to use the model in Python: |
| ``` |
| from gensim.models.doc2vec import Doc2Vec |
| |
| pat2vec_model = Doc2Vec.load('pat2vec_dim10.model') |
| |
| pat2vec_model.infer_vector(["M54.1", "J06.9", "I10.90", "R51"]) |
| ``` |
|
|
