保存预测数据

linear/train.py

@@ -59,15 +59,19 @@ def demo():
 	print("测试集MSE:",test_mse)
 
 	# train_curve = curce_data(X_train,y_train,y_pred_train)
-	# test_curve = curce_data(X_test,y_test,y_pred_test)
+	test_curve = curce_data(X_test,y_test,y_pred_test)
 	print("推广mse差", test_mse-train_mse)
 
 	'''
 	with open("train_curve.csv","w") as f :
 		f.writelines("\n".join(train_curve))
+	保存数据
 	with open("test_curve.csv","w") as f :
 		f.writelines("\n".join(test_curve))
 	'''
+	for x in test_curve:
+		print(x)
+
 	return X_train,y_train, model.coef_, model.intercept_