limit memory by using only one classifier instance

[pnews.git] / war / src / main / java / pnews / NER.java

diff --git a/war/src/main/java/pnews/NER.java b/war/src/main/java/pnews/NER.java
index bcb8951..2745868 100644 (file)
--- a/war/src/main/java/pnews/NER.java
+++ b/war/src/main/java/pnews/NER.java
@@ -6,19 +6,14 @@ import java.util.List;
 import java.util.logging.Logger;
 
 import edu.stanford.nlp.ie.crf.CRFClassifier;
-import edu.stanford.nlp.ling.CoreLabel;
+import edu.stanford.nlp.util.CoreMap;
 import edu.stanford.nlp.util.Triple;
 
 /** https://stanfordnlp.github.io/CoreNLP/api.html */
 public class NER {
         private static final String CLASS_NAME = NER.class.getName();
         private static final Logger LOG = Logger.getLogger(CLASS_NAME);
-        private static final ThreadLocal<CRFClassifier<CoreLabel>> classifier = new ThreadLocal<CRFClassifier<CoreLabel>>() {
-                @Override 
-                protected CRFClassifier<CoreLabel> initialValue() {
-                        return CRFClassifier.getDefaultClassifier();
-                }
-        };
+        private static final CRFClassifier<CoreMap> classifier = CRFClassifier.getDefaultClassifier();
         
         public static List<String> classify(String str, List<String> entities) throws ClassCastException, ClassNotFoundException, IOException {
                 
@@ -31,18 +26,20 @@ public class NER {
                 OpenNLP.classify(str, entities);
                                 
                 synchronized (classifier) {
-                        triples = classifier.get().classifyToCharacterOffsets(str);
-                        for (Triple<String, Integer, Integer> t: triples) {
-                                w = str.substring(t.second, t.third);
-                                if (!entities.contains(w))
-                                        entities.add(w);
-                        }
+                        triples = classifier.classifyToCharacterOffsets(str);
+                }
+                 
+                for (Triple<String, Integer, Integer> t: triples) {
+                        w = str.substring(t.second, t.third);
+                        if (!entities.contains(w))
+                                entities.add(w);
                 }
                 
                 entities.remove("CNET");
                 entities.remove("Read More");
                 entities.remove("New");
                 entities.remove("App");
+                entities.remove("Digital Trends");
                 
                 LOG.exiting(CLASS_NAME, FUNCTION_NAME, entities);