From ab5488c833a28f559a0fa2a12fe41b5b9527ec46 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=83=E4=B8=83?=
 <13682509+zz12345--6@user.noreply.gitee.com>
Date: Fri, 5 Jul 2024 19:38:56 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BD=AF=E4=BB=B6=E5=B7=A5=E7=A8=8B1=E7=8F=AD?=
 =?UTF-8?q?=20=E5=BC=A0=E7=BF=A0=E7=BF=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 soft/data.py    | 174 ++++++++++++++++++++++++++++++++++++++++++++++++
 soft/predict.py |   8 +++
 2 files changed, 182 insertions(+)
 create mode 100644 soft/data.py
 create mode 100644 soft/predict.py

diff --git a/soft/data.py b/soft/data.py
new file mode 100644
index 0000000..f1964a5
--- /dev/null
+++ b/soft/data.py
@@ -0,0 +1,174 @@
+datas = [
+    {
+        "city": "吕梁",
+        "area": 100,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 8000,
+    },
+    {
+        "city": "吕梁",
+        "area": 120,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 8300,
+    },
+    {
+        "city": "吕梁",
+        "area": 130,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 8600,
+    },
+    {
+        "city": "吕梁",
+        "area": 140,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9000,
+    },
+    {
+        "city": "吕梁",
+        "area": 150,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9300,
+    },
+    {
+        "city": "吕梁",
+        "area": 160,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9500,
+    },
+    {
+        "city": "吕梁",
+        "area": 170,
+        "rooms": 4,
+        "school": 3,
+        "style": 2,
+        "price": 9500,
+    },
+    {
+        "city": "吕梁",
+        "area": 180,
+        "rooms": 4,
+        "school": 4,
+        "style": 3,
+        "price": 9600,
+    },
+    {
+        "city": "吕梁",
+        "area": 190,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9400,
+    },
+    {
+        "city": "太原",
+        "area": 210,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9200,
+    },
+    {
+        "city": "太原",
+        "area": 200,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9100,
+    },
+    {
+        "city": "太原",
+        "area": 190,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 9000,
+    },
+    {
+        "city": "吕梁",
+        "area": 180,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 8900,
+    },
+    {
+        "city": "太原",
+        "area": 170,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 8800,
+    },
+    {
+        "city": "太原",
+        "area": 160,
+        "rooms": 2,
+        "school": 1,
+        "style": 1,
+        "price": 8700,
+    },
+    ]
+
+
+import numpy as np
+from datas import datas
+X=[]
+Y=[]
+cityMark={"吕梁":1,"太原":2}
+for item in datas:
+    single=[]
+
+single.append(cityMark[item["city"]])
+single.append(item["rooms"])
+single.append(item["school"])
+single.append(item["style"])
+single.append(item["area"])
+X.append(single)
+Y.append(item["price"])
+
+X=np.array(X)
+Y=np.array(Y)
+
+theta=np.linalg.pinv(X.T.dot(X)).dot(Y)
+
+
+print(theta)
+
+print(theta.dot(np.array([2,100,2,1,1])))
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
diff --git a/soft/predict.py b/soft/predict.py
new file mode 100644
index 0000000..3cae856
--- /dev/null
+++ b/soft/predict.py
@@ -0,0 +1,8 @@
+import numpy as np
+a=np.array([[3,2,3],[3,2,3],[3,2,3]])
+b=np.array([1,2,3])
+
+
+print(a.T)
+print(a.dot(b))
+print(np.linalg.pinv(a))
-- 
Gitee