git-annex in ntq8312@kibo:/data/ntq8312/ml_measurement_error_public

[ml_measurement_error_public.git] / civil_comments / get_perspective_scores.py

diff --git a/civil_comments/get_perspective_scores.py b/civil_comments/get_perspective_scores.py
index e8e542b9bec110f051e3384ad958489b9b0e31b1..1f164e7ed28ca9aa3a00f5082d1c647ba190cc2d 100644 (file)
--- a/civil_comments/get_perspective_scores.py
+++ b/civil_comments/get_perspective_scores.py
@@ -11,16 +11,17 @@ API_KEY = open('perspective_api_key').read()
 
 client = discovery.build("commentanalyzer","v1alpha",developerKey=API_KEY,discoveryServiceUrl="https://commentanalyzer.googleapis.com/$discovery/rest?version=v1alpha1",static_discovery=False,)
 
-csvreader = csv.DictReader(open("all_data.csv"))
+csvreader = csv.DictReader(open("all_data.csv"), dialect='unix')
 
 outfile = Path("perspective_results.json")
 already_scored = set()
 if outfile.exists():
     already_scored = set([json.loads(l)['id'] for l in open(str(outfile),'r')])
 
-with open("perspective_results.json",'w') as of:
+print(f"loaded {len(already_scored)} scored comments")
+with open("perspective_results.json",'a') as of:
     for line in csvreader:
-        if line['id'] not in already_scored:
+        if line['id'] not in already_scored and len(line.get('comment_text','')) > 0:
             analyze_request = {'comment':{'text':line['comment_text']},
                                'languages':['en'],
                                'requestedAttributes':{'TOXICITY':{},