面积与价格
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from sklearn.linear_model import LinearRegression
data = np.genfromtxt("house_data.txt", delimiter=",")
x_data = data[:,0]
y_data = data[:,1]
x_data = data[:,0,np.newaxis]
y_data = data[:,1,np.newaxis]
# 创建并拟合模型
model = LinearRegression()
from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test = train_test_split(x_data,y_data,test_size=0.8)
model.fit(x_train,y_train)
a = model.coef_
b = model.intercept_
print('模型为：y={}x-{}'.format(a[0][0],-b[0]))
# 画图
plt.plot(x_data, y_data, 'b.')
plt.plot(x_data, model.predict(x_data), 'r')
y_predict = model.predict(x_test[:1])
print(y_test[:1])
print(y_predict)
plt.show()
