Skip to content
Snippets Groups Projects
Commit b520612c authored by Asia Belfiore's avatar Asia Belfiore
Browse files

Initial Commit

parents
No related branches found
No related tags found
No related merge requests found
.DS_Store 0 → 100644
File added
FROM ubuntu:jammy
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
\ No newline at end of file
package:
tar --exclude=coursework1.tar.gz -cvzf coursework1.tar.gz .
\ No newline at end of file
%% Cell type:markdown id: tags:
Belfiore Asia, CID:02129867
MSc Advanced Computing, Spring Term, January 2025
# **Acute Kidney Injury (AKI) Neural Network Predictor**
%% Cell type:code id: tags:
``` python
import numpy as np # add to requirements.txt
import torch # add to requirements.txt
import torch.nn as nn
training_data_path = '/data/training.csv'
test_data_path = '/data/test.csv'
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.fc1 = nn.Linear(2, 2)
self.fc2 = nn
```
This diff is collapsed.
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
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