reflexive 1.0.14__tar.gz → 1.0.16__tar.gz

{reflexive-1.0.14/reflexive.egg-info → reflexive-1.0.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: reflexive
-Version: 1.0.14
+Version: 1.0.16
 Summary: Supports AWS Reflexive Expressions Analysis
 Home-page: https://github.com/nlytx/reflexive
 Author: Andrew Gibson

{reflexive-1.0.14 → reflexive-1.0.16}/reflexive/visualise.py

@@ -6,8 +6,36 @@ class Display:
     config:cfg.Config = None
     
     defaults = {
-        "priority_tags": ["AR","EP","VR_EV_CN","ER_AF","RR","KP"],
-        "colours": {"VR_EV_CN": "#ff6644","ER_AF": "#dd44cc","AR": "#00cc00","EP": "#aacc33","RR": "#00aaff","KP":"#aaaacc"}}
+        "priority_tags": [
+            "VR_ER",
+            "AR",
+            "RR",
+            "EP_EV",
+            "AF_CN",
+            "EO",
+            "EA"
+        ],
+        "colours": {
+            "VR_ER": "#ff6644",
+            "AR": "#00cc00",
+            "RR": "#6699ff",
+            "EP_EV": "#aacc33",
+            "AF_CN": "#dd44cc",
+            "EO":"#cccc00",
+            "EA":"#33cccc"
+        }
+    }
+    
+    # "VR_ER": "#ff6644", #Orange
+	# 	"AR": "#00cc00", #Green
+	# 	"RR": "#6699ff", #Blue
+	# 	"EP_EV": "#aacc33", #Lime
+	# 	"AF_CN": "#dd44cc", #magenta
+	# 	"EO":"#cccc00", #dark yellow 
+	# 	"EA":"#33cccc" #dark Cyan
+    # {
+    #     "priority_tags": ["VR_ER","AR","RR","EP_EV","AF_CN","KP"],
+    #     "colours": {"VR_EV_CN": "#ff6644","ER_AF": "#dd44cc","AR": "#00cc00","EP": "#aacc33","RR": "#00aaff","KP":"#aaaacc"}}
     
     def __init__(self,aws):
         self.aws = aws
@@ -18,7 +46,7 @@ class Display:
     def set_default_parameters(self):
         priority_tags = self.defaults['priority_tags']
         colours = self.defaults['colours']
-        options = {"ents": list(colours.keys()), "colors": list(colours.values())}
+        options = {"ents": list(colours.keys()), "colors": colours}
         self.config.set_display_parameters(priority_tags,colours,options)
        
      
@@ -32,9 +60,15 @@ class Display:
         temp_df['keyphrase_offsets'] = temp_df.KeyPhraseResults.apply(self.collect_keyphrase_offsets)
         return temp_df
     
+    def add_syntax_offsets(self,df):
+        temp_df = df.copy()
+        temp_df['syntax_offsets'] = temp_df.SyntaxResults.apply(self.collect_syntax_offsets)
+        return temp_df
+    
     def add_offsets(self,df):
         df = self.add_reflexive_offsets(df)
-        return self.add_keyphrase_offsets(df)
+        df = self.add_keyphrase_offsets(df)
+        return self.add_syntax_offsets(df)
     
     def create_displacy(self,df):
         all_ents = list(df.apply(self.render_record,axis=1))
@@ -53,12 +87,15 @@ class Display:
         text = record['text']
         reflexive_offsets = record['reflexive_offsets']
         keyphrase_offsets = record['keyphrase_offsets']
+        syntax_offsets = record['syntax_offsets']
         ents = []
         taken = []
         offsets = []
         for tag in tags:
             if tag in reflexive_offsets:
                 offsets = reflexive_offsets[tag]
+            elif tag in syntax_offsets:
+                offsets = syntax_offsets[tag]
             elif tag in keyphrase_offsets:
                 offsets = keyphrase_offsets[tag]
             
@@ -100,11 +137,24 @@ class Display:
         for rr in rrs:
             if rr['Score']>0.5:
                 ent_type = rr['Type']
-                if ent_type in ['VR','EV','CN']:
-                    label = "VR_EV_CN"
-                elif ent_type in ['ER','AF']:
-                    label = "ER_AF"
+                if ent_type in ['VR','ER']:
+                    label = "VR_ER"
+                elif ent_type in ['EP','EV']:
+                    label = "EP_EV"
+                elif ent_type in ['CN','AF']:
+                    label = "AF_CN"
                 else:
                     label = ent_type
                 new_rrs.setdefault(label,[]).append((rr['BeginOffset'],rr['EndOffset']))
-        return new_rrs
+        return new_rrs
+    
+    def collect_syntax_offsets(self,syntax_results):
+        offsets = {}
+        for sr in syntax_results:
+            pos = sr['PartOfSpeech']
+            if pos['Score']>0.99:
+                if pos['Tag'] in ['PRON','NOUN','ADJ']:
+                    offsets.setdefault("EO",[]).append((sr['BeginOffset'],sr['EndOffset']))
+                if pos['Tag'] in ['ADV','VERB']:
+                    offsets.setdefault("EA",[]).append((sr['BeginOffset'],sr['EndOffset']))
+        return offsets

{reflexive-1.0.14 → reflexive-1.0.16/reflexive.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: reflexive
-Version: 1.0.14
+Version: 1.0.16
 Summary: Supports AWS Reflexive Expressions Analysis
 Home-page: https://github.com/nlytx/reflexive
 Author: Andrew Gibson

{reflexive-1.0.14 → reflexive-1.0.16}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = reflexive
-version = 1.0.14
+version = 1.0.16
 author = Andrew Gibson
 author_email = andrew@nlytx.io
 description = Supports AWS Reflexive Expressions Analysis