Add linear_regression jupyter notebook.

zjucsxxd · Nov 5, 2018 · 40ddaf7 · 40ddaf7
1 parent d51a5c5
commit 40ddaf7
Show file tree

Hide file tree

Showing 4 changed files with 312 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 .idea
 env
+.ipynb_checkpoints
diff --git a/data/house-prices.csv b/data/house-prices.csv
@@ -1,3 +1,4 @@
+Size,Rooms,Price
 2104,3,399900
 1600,3,329900
 2400,3,369000

diff --git a/src/linear_regression/demo.py b/src/linear_regression/demo.py
@@ -7,7 +7,7 @@
 from linear_regression import LinearRegression
 
 # Load the data.
-DATA = np.genfromtxt('./data/house-prices.csv', delimiter=',')
+DATA = np.genfromtxt('./data/house-prices.csv', delimiter=',', skip_header=True)
 
 # Split the by input and output.
 X = DATA[:, 0:2]
@@ -17,7 +17,7 @@
 LINEAR_REGRESSION = LinearRegression(X, Y)
 
 # Train linear regression.
-NUM_ITERATIONS = 30
+NUM_ITERATIONS = 50
 LAMBDA_PARAM = 0
 ALPHA = 0.1
 
@@ -29,9 +29,6 @@
     COST_HISTORY
 ) = LINEAR_REGRESSION.train(ALPHA, LAMBDA_PARAM, NUM_ITERATIONS)
 
-print(np.min(X[:, 0]))
-print(np.min(X[:, 1]))
-
 print('Initial cost: {0}\n'.format(COST_HISTORY[0]))
 print('Optimized cost: {0}\n'.format(COST_HISTORY[-1:]))
 

diff --git a/src/linear_regression/linear_regression.ipynb b/src/linear_regression/linear_regression.ipynb