diff --git a/demo7.5.txt b/demo7.5.txt new file mode 100644 index 0000000000000000000000000000000000000000..30a4cc366ca88538456c2310ce1349f8d67bff79 --- /dev/null +++ b/demo7.5.txt @@ -0,0 +1,24 @@ +import numpy as np +#a=np.array([[1,2,3],[1,2,3],[1,2,3]]) +#b=np.array([1,2,3]) +#print(a.dot(b))#点乘 +#print(a.T)#转置 +#print(np.linalg.pinv(a))#逆 +from datas import datas +X=[] +Y=[] +cityMark={"吕梁":1,"太原":2} +for item in datas: + single=[] + single.append(cityMark[item["city"]]) + single.append(item["area"]) + single.append(item["rooms"]) + single.append(item["school"]) + single.append(item["style"]) + X.append(single) + Y.append(item["price"]) +X=np.array(X) +Y=np.array(Y) + +theta=np.linalg.pinv(X.T.dot(X)).dot(X.T).dot(Y) +print(theta.dot(np.array([2,100,2,1,2]))) \ No newline at end of file