Skip to content
Snippets Groups Projects
Commit 2944623c authored by Rahul Khorana's avatar Rahul Khorana
Browse files

got spec from website

parents
No related branches found
No related tags found
1 merge request!1got spec from website
FROM ubuntu:oracular
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get -yq install python3-pip
COPY requirements.txt /model/
RUN pip3 install -r /model/requirements.txt
COPY model.py /model/
CMD /model/model.py --input=/data/test.csv --output=/data/aki.csv
model.py 0 → 100755
#!/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()
\ No newline at end of file
This diff is collapsed.
This diff is collapsed.
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