From 4537bdbebb5fe64c50080e7874d407f10a0676b7 Mon Sep 17 00:00:00 2001
From: Charles <sircharlesaze@gmail.com>
Date: Thu, 23 Jan 2020 09:57:39 +0100
Subject: WIP: CLI interface interact with Model class, subprogram call Model
 methods to satisfy the subject

---
 src/train.py | 30 ++----------------------------
 1 file changed, 2 insertions(+), 28 deletions(-)

(limited to 'src/train.py')

diff --git a/src/train.py b/src/train.py
index a31d032..0a68916 100644
--- a/src/train.py
+++ b/src/train.py
@@ -1,32 +1,6 @@
-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
+from model import Model
 
 
 if __name__ == "__main__":
-    m = Model()
-    m.read_data()
-    m.normalize_data()
+    m = Model(datafilename="../data.csv", thetafilename="./theta")
     m.train()
-    m.write_theta()
-- 
cgit