Hierarchy-Transformers
/

HiT-MPNet-WordNetNoun

Feature Extraction

hierarchy-transformers

hierarchy-encoding

subsumption-relationships

Model card Files Files and versions

lawhy commited on Jan 21, 2024

Commit

5b3b79d

·

verified ·

1 Parent(s): 350c28e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ gpu_id = 0
 device = get_torch_device(gpu_id)
 # load the model
-model = HierarchyTransformer.load_pretrained('Hierarchy-Transformers/HiT-MiniLM-L12-WordNet', device)
 # entity names to be encoded.
 entity_names = ["computer", "personal computer", "fruit", "berry"]

 device = get_torch_device(gpu_id)
 # load the model
+model = HierarchyTransformer.load_pretrained('Hierarchy-Transformers/HiT-MPNet-WordNetNoun', device)
 # entity names to be encoded.
 entity_names = ["computer", "personal computer", "fruit", "berry"]