bucketing will happen in chicago-justice i think

chicago-justice-project · Nov 6, 2019 · c6ae60f · c6ae60f
1 parent e831faa
commit c6ae60f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/lib/tagnews/senteval/eval.py b/lib/tagnews/senteval/eval.py
@@ -2,6 +2,8 @@
 from google.cloud.language import enums
 from google.cloud.language import types
 
+from tagnews.senteval import police_words
+
 
 def process_google_result(text):
     document = types.Document(content=text, type=enums.Document.Type.PLAIN_TEXT)
@@ -10,7 +12,7 @@ def process_google_result(text):
     for entity in sentiment.entities:
         clean_entity = "".join(filter(str.isalpha, entity)).lower()
 
-        if clean_entity in ["police", "officer", "cop", "officers", "pigs"]:
+        if clean_entity in police_words:
 
             for mention in entity.mentions:
                 return mention.sentiment.score
@@ -66,8 +68,6 @@ def is_police_entity(self, entity):
                 return entity
             return False
 
-    def set_bucket_value(self):
-        pass
 
 def pre_process_text(html_text):
     """

diff --git a/lib/tagnews/senteval/police_words.py b/lib/tagnews/senteval/police_words.py
@@ -0,0 +1 @@
+police_words = ["police", "officer", "cop", "officers", "pigs"]
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		police_words = ["police", "officer", "cop", "officers", "pigs"]