Skip to content
Snippets Groups Projects
model.py 537 B
Newer Older
Rahul Khorana's avatar
Rahul Khorana committed
#!/usr/bin/env python3

import argparse
import csv
import random

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--input", default="test.csv")
    parser.add_argument("--output", default="aki.csv")
    flags = parser.parse_args()
    r = csv.reader(open(flags.input))
    w = csv.writer(open(flags.output, "w"))
    w.writerow(("aki",))
    next(r) # skip headers
    for _ in r:
        # TODO: Implement a better model
        w.writerow((random.choice(["y", "n"]),))

if __name__ == "__main__":
    main()