|
|
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from sklearn import datasets,linear_model
def get_data('input_data'):
cnames=['square_feet','price']
data=pd.read_csv('input_data.csv',header=2,names=cnames,encoding='utf-8')
X_parameter=[]
Y_parameter=[]
for single_square_feet,single_price_value in zip(data['square_feet'],data['price']):
X_parameter.append([float(single_square_feet)])
Y_parameter.append(float(single_price_value))
return X_parameter,Y_parameter
def linear_model_main(X_parameter,Y_parameter,predict_value):
linear=linear_model.LinearRegression()
linear.fit(X_parameter,Y_parameter)
predict_outcome=linear.predict(predict_value)
predictions={}
predictions['intercept']=linear.intercept_
predictions['coefficient']=linear.coef_
predictions['predicted_value']=predict_outcome
return predictions
X,Y=get_data('input_data.csv')
predictvalue=700
result=linear_model_main(X,Y,predictvalue)
print ("Intercept value",result['intercept'])
print ("coefficient",result['coefficient'])
print ("Predicted value:",result['predicted_value'])
在返回X_parameter，Y_parameter的时候就有问题了请下帮忙看下，最好能给出正确代码，，谢谢
|
|
