Sentence grouping on NER replace

Author: alexsands

plasticity/sapien/core.py

@@ -246,14 +246,16 @@ def ner_replace(self):
         ner = self.ner()
         data_out = []
         for sg_data, sg_ner in zip(self.data, ner):
+            sentence_group_out = []
             for alt_data, alt_ner in zip(sg_data.alternatives, sg_ner):
                 # Replace the text in the sentence with the named entities
                 new_sentence = alt_data.sentence
                 for graph in alt_ner:
                     for entity in graph:
                         new_sentence = new_sentence.replace(
                             entity['entity'], entity['ner'][0]['label'])
-                data_out.append(new_sentence)
+                sentence_group_out.append(new_sentence)
+            data_out.append(sentence_group_out)
         return data_out
 
 

plasticity/version.py

@@ -1,2 +1,2 @@
-__version_info__ = ('0', '0', '5')
+__version_info__ = ('0', '0', '6')
 __version__ = '.'.join(__version_info__)