# -*- coding: utf-8 -*-
"""
Created on Mon Oct  8 14:02:28 2018

@author: Administrator
"""

import os
dir(os)
os.chdir(r'C:\Users\Administrator\Desktop\10-8\04-回归分析\data')
from sklearn.linear_model import LinearRegression
import numpy as np
X=np.array([ [0,0],[1,1],[2,2]]) 
y=np.array([0,1,2])
clf=LinearRegression().fit(X,y)
clf.predict([[4,3]])
clf.get_params()
print(clf)


from sklearn.datasets import load_boston
boston=load_boston()
import pandas as pd
data=pd.DataFrame(boston.data)
target=boston.target

m=LinearRegression().fit(data.iloc[:,5:6],target)
pre =m.predict(data.iloc[:,5:6])

import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif']='SimHei'
plt.rcParams['axes.unicode_minus']=False
plt.figure(figsize=(8,8))
plt.scatter(data.iloc[:,5:6],target)
plt.plot(data.iloc[:,5:6],pre,color='r')
plt.title('线性回归')
plt.ylabel('房价')
plt.xlabel('平均房间数')
plt.show()