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 index 9bce260..67dd963 100644 --- 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 entityToTermFrequencyInfo(List 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(ev.getConceptInfo().getSemanticTypeSet()), tupleSet, entity.getFieldId() == null ? false : @@ -447,7 +448,7 @@ public List entityToTermFrequencyInfo(List entityList) { cui, 1, ev.getScore(), - getTreecodes(conceptString))); + getTreecodes(preferredName))); } } }