Fixed error in MMI treecode output

Changed treecode lookup to use concept preferred name instead of synonym.
LHNCBC · Apr 5, 2020 · d6716b2 · d6716b2
1 parent a164b03
commit d6716b2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/gov/nih/nlm/nls/metamap/lite/resultformats/mmi/MMI.java b/src/main/java/gov/nih/nlm/nls/metamap/lite/resultformats/mmi/MMI.java
@@ -436,8 +436,9 @@ public List<TermFrequency> entityToTermFrequencyInfo(List<Entity> entityList) {
 				   entity.isNegated() ? 1 : 0, // neg?
 				   posInfo); 
 	  tupleSet.add(tuple);
+	  String preferredName = ev.getConceptInfo().getPreferredName();
 	  termFreqMap.put(tfKey,
-			  new TermFrequency(ev.getConceptInfo().getPreferredName(),
+			  new TermFrequency(preferredName,
 					    new ArrayList<String>(ev.getConceptInfo().getSemanticTypeSet()),
 					    tupleSet, 
 					    entity.getFieldId() == null ? false :
@@ -447,7 +448,7 @@ public List<TermFrequency> entityToTermFrequencyInfo(List<Entity> entityList) {
 					    cui,
 					    1,
 					    ev.getScore(),
-					    getTreecodes(conceptString)));
+					    getTreecodes(preferredName)));
 	}
       }
     }