Skip to content
Snippets Groups Projects
Commit bdec1898 authored by Vaggelis Atlidakis's avatar Vaggelis Atlidakis
Browse files

Do not replicate two_year_recid field

I assume you don't want the second file, compas-scores-two-years.csv, to contain
the column two_year_recid twice.
parent 407cb07f
No related branches found
No related tags found
No related merge requests found
...@@ -177,6 +177,7 @@ def is_race(race): ...@@ -177,6 +177,7 @@ def is_race(race):
def write_two_year_file(f, pop, test, headers): def write_two_year_file(f, pop, test, headers):
headers = list(headers)
headers.append('two_year_recid') headers.append('two_year_recid')
with open(f, 'w') as o: with open(f, 'w') as o:
writer = DictWriter(o, fieldnames=headers) writer = DictWriter(o, fieldnames=headers)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment