From 7420a3c140a8038d0a16af27f11419ef0e9e99ef Mon Sep 17 00:00:00 2001 From: Moon <14648512+Moon_777@user.noreply.gitee.com> Date: Fri, 5 Jul 2024 11:03:34 +0000 Subject: [PATCH] =?UTF-8?q?=E8=BD=AF=E4=BB=B6=E5=B7=A5=E7=A8=8B2=E7=8F=AD?= =?UTF-8?q?=E2=80=94=E4=BE=AF=E4=BD=A9=E7=91=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Moon <14648512+Moon_777@user.noreply.gitee.com> --- datas.py | 140 +++++++++++++++++++++++++++++++++++++++++++++++++++++ predict.py | 17 +++++++ 2 files changed, 157 insertions(+) create mode 100644 datas.py create mode 100644 predict.py diff --git a/datas.py b/datas.py new file mode 100644 index 0000000..68e0f43 --- /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 0000000..ce310a1 --- /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 -- Gitee