fixed a bug in the semantic module when using the new ontology

angelosalatino · Jul 30, 2019 · 96ea777 · 96ea777
1 parent f7e8f17
commit 96ea777
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/classifier/semanticmodule.py b/classifier/semanticmodule.py
@@ -162,7 +162,7 @@ def classify_semantic(self):
                     sim   = topic_item["sim_w"]
 
 
-                    if m >= min_similarity:
+                    if m >= min_similarity and topic in self.cso["topics_wu"]:
 
 
                         if topic in found_topics:
@@ -227,6 +227,8 @@ def classify_semantic(self):
         unique_topics = {}
         for tp,topic in found_topics.items():
             prim_label = self.get_primary_label(tp,self.cso["primary_labels_wu"])
+            if prim_label == 'network_structures':
+                print('Here I found you:', tp)
             if prim_label in unique_topics:
                 if unique_topics[prim_label] < topic["score"]:
                     unique_topics[prim_label] = topic["score"]