reflexive 1.0.15__py3-none-any.whl → 1.0.16__py3-none-any.whl

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
@@ -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.15.dist-info → reflexive-1.0.16.dist-info}/METADATA

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

{reflexive-1.0.15.dist-info → reflexive-1.0.16.dist-info}/RECORD

@@ -3,10 +3,10 @@ reflexive/analyse.py,sha256=UzWwgjAFNjeWFkCQ2o99g2vWajf17_OtSq4dFCvuPYU,17489
 reflexive/cfg.py,sha256=Ges35G234P2lvOQHgPZQae5hMSOGyBsmp1bY_yQEKkk,4303
 reflexive/session.py,sha256=MbqwTsYTgq_e_gw3mb1eRv6USs-zZ2cTCrvUNWuKfAQ,10067
 reflexive/util.py,sha256=WQ1oyzDi1i8wQ6IBwBPk6IFy07YKhg-Ug2FsOGVJRJQ,3649
-reflexive/visualise.py,sha256=gsTqZRRvmfonxz6ZWFhaZURl9LalRMWrSZBrpasIitQ,4031
-reflexive-1.0.15.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-reflexive-1.0.15.dist-info/LICENSE.txt,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-reflexive-1.0.15.dist-info/METADATA,sha256=dC2sVG_Bv-h3PS7opJeC2alRO9PWdfEdDfRk5eJrSvw,12038
-reflexive-1.0.15.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-reflexive-1.0.15.dist-info/top_level.txt,sha256=pOMr-QGleRBRCFBozgvM-UUUmOjD_-naJfu1522E2V8,10
-reflexive-1.0.15.dist-info/RECORD,,
+reflexive/visualise.py,sha256=9f-4tfUsv2-zr9tlKZcEEkugEPffcoAogXMWqnmcECs,5615
+reflexive-1.0.16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+reflexive-1.0.16.dist-info/LICENSE.txt,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+reflexive-1.0.16.dist-info/METADATA,sha256=_Ck3ePqPaq4QWZzkmqAnnZqCU1eTZXFZeLG7Lb2twg4,12038
+reflexive-1.0.16.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+reflexive-1.0.16.dist-info/top_level.txt,sha256=pOMr-QGleRBRCFBozgvM-UUUmOjD_-naJfu1522E2V8,10
+reflexive-1.0.16.dist-info/RECORD,,