def CART(parameters): csv = pd.read_csv('static/data/processedDataForCalculate.csv', header=None, delimiter=',') x = csv.iloc[:, 0:-1].values # '区','所处楼层级别','总楼层数','建造年份','面积','朝向','是否近地铁','挂牌年','挂牌月','挂牌日','纬度','经度' y = csv.iloc[:, -1].values # '总价' cart = tree.DecisionTreeRegressor(random_state=20, max_depth=10) cart.fit(x, y) parameters = np.array(parameters).reshape(1, -1) return cart.predict(parameters)
时间: 2024-01-10 14:03:31 浏览: 123
Import-from-Text-or-CSV.zip_csv_reading
这段代码使用 CART 决策树算法进行回归分析。首先,它加载一个 CSV 文件,该文件包含数据集。然后,它将数据集分为输入变量和目标变量,其中输入变量是数据集中的前 n-1 列,目标变量是数据集的最后一列。接下来,它使用 `tree.DecisionTreeRegressor()` 函数构建了一个 CART 回归树模型,并使用 `fit()` 函数将输入变量和目标变量拟合到模型中。最后,它将函数输入的参数 `parameters` 转换成 numpy 数组的形式,并使用 `predict()` 函数返回预测结果。
阅读全文