XGBClassifier函数怎么使用
时间: 2024-04-08 20:35:00 浏览: 78
`XGBClassifier`是XGBoost库中用于分类任务的类。下面是一个使用`XGBClassifier`函数的示例代码:
```python
from xgboost import XGBClassifier
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 创建一个分类数据集
X, y = make_classification(n_samples=100, n_features=10, random_state=1)
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=1)
# 创建XGBoost分类器对象
xgb_classifier = XGBClassifier()
# 在训练集上训练分类器
xgb_classifier.fit(X_train, y_train)
# 在测试集上进行预测
y_pred = xgb_classifier.predict(X_test)
# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print("准确率:", accuracy)
```
在上述代码中,我们首先使用`make_classification`函数创建了一个简单的分类数据集。然后,使用`train_test_split`函数将数据集划分为训练集和测试集。接下来,创建了一个`XGBClassifier`对象。然后,使用训练集对分类器进行训练,然后在测试集上进行预测,并计算准确率。
你可以根据自己的需求和数据集来调整代码中的参数和数据,以便使用`XGBClassifier`进行分类任务。
阅读全文