Moved notebook to notebook/, Added predict.py and train.py template

1 files changed, 32 insertions, 0 deletions

diff --git a/src/train.py b/src/train.py
new file mode 100644
index 0000000..a31d032
--- /dev/null
+++ b/src/train.py
@@ -0,0 +1,32 @@
+class Model:
+    def __init__(self, filename='../data.csv'):
+        self.datafile = filename
+
+    def train(self):
+        pass
+
+    def partial_theta1(self):
+        pass
+
+    def partial_theta0(self):
+        pass
+
+    def gradient_descent(self):
+        pass
+
+    def read_data(self):
+        pass
+
+    def normalize_data(self):
+        pass
+
+    def write_theta(self):
+        pass
+
+
+if __name__ == "__main__":
+    m = Model()
+    m.read_data()
+    m.normalize_data()
+    m.train()
+    m.write_theta()