Browse Source

活动销量线性回归

yufeng0528 4 years ago
parent
commit
639fffecf8
1 changed files with 4 additions and 4 deletions
  1. 4 4
      linear/train.py

+ 4 - 4
linear/train.py

@@ -31,8 +31,8 @@ def read_data(path):
31 31
 
32 32
 
33 33
 def demo():
34
-	X_train,y_train=read_data("train_data")
35
-	X_test,y_test=read_data("test_data")
34
+	X_train,y_train=read_data("../bbztx/train_data")
35
+	X_test,y_test=read_data("../bbztx/test_data")
36 36
 
37 37
 	#一个对象,它代表的线性回归模型,它的成员变量,就已经有了w,b. 刚生成w和b的时候 是随机的
38 38
 	model = LinearRegression()
@@ -70,7 +70,7 @@ def demo():
70 70
 
71 71
 
72 72
 def draw_line():
73
-	x_train, y_train = read_data("train_data")
73
+	x_train, y_train = read_data("../bbztx/train_data")
74 74
 	print(x_train.tolist())
75 75
 	print(y_train.tolist())
76 76
 	draw_util.drawScatter(x_train.tolist(), y_train.tolist())
@@ -83,6 +83,6 @@ if __name__ == '__main__':
83 83
 	q = [i[0] for i in q.tolist()]
84 84
 	w = w[0]
85 85
 	b = b[0]
86
-	draw_util.drawScatterAndLine(p, q, w, b)
86
+	# draw_util.drawScatterAndLine(p, q, w, b)
87 87
 
88 88