gpt4 book ai didi

Pyspark 模型的 Python 等价物

转载 作者:行者123 更新时间:2023-12-04 14:03:59 27 4
gpt4 key购买 nike

哪个模型是“GBTRegressor”Pyspark 模型的 Python 等效模型?

简要背景:我正在尝试将 pyspark 模型重新创建为 python 模型。现有管道中使用的模型是 GBTRegressor。我想知道哪个模型是 python 的等价物,这样我就可以使用类似的参数并将模型部署到 python 上。

最佳答案

GBTRegressor 是用于回归的梯度提升树。如果你想在 python 中实现它们,你可以使用 sklearn's implementation .以下是对糖尿病数据集使用 GBT 回归的简单示例:

import matplotlib.pyplot as plt
import numpy as np
from sklearn import datasets, ensemble
from sklearn.inspection import permutation_importance
from sklearn.metrics import mean_squared_error
from sklearn.model_selection import train_test_split

# loading Data
diabetes = datasets.load_diabetes()
X, y = diabetes.data, diabetes.target

X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.1, random_state=13)

params = {'n_estimators': 500,
'max_depth': 4,
'min_samples_split': 5,
'learning_rate': 0.01,
'loss': 'ls'}

# fitting model

reg = ensemble.GradientBoostingRegressor(**params)
reg.fit(X_train, y_train)

mse = mean_squared_error(y_test, reg.predict(X_test))
print("The mean squared error (MSE) on test set: {:.4f}".format(mse))

关于Pyspark 模型的 Python 等价物,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/69102114/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com