diff --git a/datas.py b/datas.py new file mode 100644 index 0000000000000000000000000000000000000000..68e0f43cc8077e2b71e73a3066b8e32863ef9bc4 --- /dev/null +++ b/datas.py @@ -0,0 +1,140 @@ +datas=[ + { + "city":"吕梁", + "area":100, + "rooms":2, + "school":1, + "style":1, + "price":8000 + }, + { + "city": "吕梁", + "area": 100, + "rooms": 2, + "school": 1, + "style": 1, + "price": 7800 + }, + { + "city": "吕梁", + "area": 135, + "rooms": 3, + "school": 1, + "style": 1, + "price": 8300 + }, + { + "city":"吕梁", + "area":110, + "rooms":2, + "school":1, + "style":1, + "price":8200 + }, + { + "city": "吕梁", + "area": 130, + "rooms": 3, + "school": 2, + "style": 2, + "price": 6500 + }, + { + "city": "吕梁", + "area": 140, + "rooms": 3, + "school": 1, + "style": 1, + "price": 8600 + }, + { + "city": "太原", + "area": 110, + "rooms": 2, + "school": 1, + "style": 1, + "price": 8200 + }, + { + "city": "太原", + "area": 130, + "rooms": 3, + "school": 2, + "style": 2, + "price": 6500 + }, + { + "city": "太原", + "area": 140, + "rooms": 3, + "school": 1, + "style": 1, + "price": 8600 + }, +] +datas=[ + { + "city":"吕梁", + "area":100, + "rooms":2, + "school":1, + "style":1, + "price":8000 + }, + { + "city": "吕梁", + "area": 100, + "rooms": 2, + "school": 1, + "style": 1, + "price": 7800 + }, + { + "city": "吕梁", + "area": 135, + "rooms": 3, + "school": 1, + "style": 1, + "price": 8300 + }, + { + "city":"吕梁", + "area":110, + "rooms":2, + "school":1, + "style":1, + "price":8200 + }, + { + "city": "吕梁", + "area": 130, + "rooms": 3, + "school": 2, + "style": 2, + "price": 6500 + }, + { + "city": "吕梁", + "area": 140, + "rooms": 3, + "school": 1, + "style": 1, + "price": 8200 + }, + { + "city": "太原", + "area": 130, + "rooms": 3, + "school": 2, + "style": 2, + "price": 6500 + }, + { + "city": "太原", + "area": 140, + "rooms": 3, + "school": 1, + "style": 1, + "price": 8600 + }, +] \ No newline at end of file diff --git a/predict.py b/predict.py new file mode 100644 index 0000000000000000000000000000000000000000..ce310a10f80e27460054df6f624f99dc7b77bfae --- /dev/null +++ b/predict.py @@ -0,0 +1,17 @@ +import signal +import numpy as np +from datas import datas +X=[] +Y=[] +cityMark={"吕梁":1,"太原":2} +for item in datas: + signal=[] + signal.append(cityMark[item["city"]]) + signal.append(item["area"]) + signal.append(item["rooms"]) + signal.append(item["school"]) + signal.append(item["style"]) + X.append(signal) + Y.append(item["price"]) +theta=np.linalg.pinv(X.T.dot(X)).dot(X.T).dot(Y) + print(theta.dot(np.array([2,100,2,1,1]))) \ No newline at end of file