zidatalab commited on
Commit
6617157
·
1 Parent(s): f1c5f97

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
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
+ ```