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"])