我正试图建立一个房价预测模型。我使用python学习工具集。我使用的数据集有总楼面面积和公寓位置,我已将其转换为虚拟特征。因此,数据集看起来如下:
然后,我建立了一个学习曲线,看看模型做得如何。我用这种方式建立了学习曲线:
from matplotlib import pyplot as plt
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import learning_curve
model = LinearRegression()
training_sizes, train