diff --git a/7.5/__pycache__/datas.cpython-312.pyc b/7.5/__pycache__/datas.cpython-312.pyc new file mode 100644 index 0000000000000000000000000000000000000000..e75cd804250d35e30bfd5e4a63407b8fb4dd3063 Binary files /dev/null and b/7.5/__pycache__/datas.cpython-312.pyc differ diff --git a/7.5/datas.py b/7.5/datas.py new file mode 100644 index 0000000000000000000000000000000000000000..dc4384ee7a0fd4da4903e6ccc55ccec5697f4928 --- /dev/null +++ b/7.5/datas.py @@ -0,0 +1,58 @@ +datas=[ +{ + "city":"吕梁", + "area":"100", + "rooms":"2", + "school":"1", + "style":"1", + "price":8000, +}, +{ + "city":"吕梁", + "area":"100", + "rooms":"2", + "school":"1", + "style":"1", + "price":8000, +}, +{ + "city":"太原", + "area":"100", + "rooms":"2", + "school":"1", + "style":"1", + "price":8000, +}, +{ + "city":"吕梁", + "area":"200", + "rooms":"2", + "school":"1", + "style":"0", + "price":9000, +}, +{ + "city":"吕梁", + "area":"100", + "rooms":"2", + "school":"1", + "style":"1", + "price":8000, +}, +{ + "city":"吕梁", + "area":"100", + "rooms":"2", + "school":"1", + "style":"1", + "price":8000, +}, +{ + "city":"吕梁", + "area":"100", + "rooms":"2", + "school":"1", + "style":"1", + "price":8000, +} +] \ No newline at end of file diff --git a/7.5/predict.py b/7.5/predict.py new file mode 100644 index 0000000000000000000000000000000000000000..ce599595853ef371aa098a083a5d7581dcaf012b --- /dev/null +++ b/7.5/predict.py @@ -0,0 +1,29 @@ +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["area"]) + single.append(item["rooms"]) + single.append(item["school"]) + single.append(item["style"]) + + X.append(single) + Y.append(item["price"]) + +X = np.array(X, dtype=float) +Y = np.array(Y, dtype=float) +print(X) +print(Y) +theta = np.linalg.pinv(X.T.dot(X)).dot(X.T).dot(Y) +print(theta) + +prediction = theta.dot(np.array([2, 100, 2, 1, 1])) +print(prediction)