1 files changed, 2 insertions, 28 deletions
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()