【機械学習初心者】部屋の広さから家賃を予測

はじめに

機械学習の知識を使って財務シュミレーションを簡潔に行いたいと思いPythonで実装してみました。

対象者

• 機械学習初心者
• 財務シュミレーションなど行いたい方

開発した時の環境

version 
Python 3.8.5
Homebrew 3.0.2 
jupyter notebook

単回帰分析の実装

部屋の広さx家賃がy万円

まず取り込みたいCSVデータをインポートします

import pandas as pd
df = pd.read_csv('sample.csv')

グラフ描画ツールMatplotlib

import matplotlib.pyplot as plt

xとyの散布図プロット

plt.scatter(x, y)
plt.show()

#予測値
xx = x*x
xy = x*y
a = xy.sum()/xx.sum()

plt.scatter(x,y, label='y')
#学習フェーズ
plt.plot(x, a*x, label='y_hat', color ='red')
plt.legend()
plt.show

#予測値
x_new = 1000 #○○平米の部屋
mean = df.mean()
mean['x']

xc = x_new - mean['x']
xc 

yc = a*xc

y_hat = a*xc + mean['y']


def predict(x):

 a = 30069.022519284063
 xm = 37.622219999999999
 ym = 121065.0
 xc = x - xm 
 y_hat = a*xc + ym

 return y_hat