随机森林(Random Forest)是一种强大且常用于分类和回归任务的机器学习算法。它是基于决策树的集成学习方法,通过构建多个决策树,并取其结果的平均值或多数投票来提高模型的性能和鲁棒性。以下是使用 scikit-learn 库构建随机森林分类器的基本步骤:

步骤:

1. 导入库:
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score, classification_report, confusion_matrix

2. 准备数据:
# 读取数据集
data = pd.read_csv('your_dataset.csv')

# 分割特征和标签
X = data.drop('target_column', axis=1)
y = data['target_column']

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

3. 选择模型并训练:
# 选择随机森林分类器
model = RandomForestClassifier(n_estimators=100, random_state=42)

# 训练模型
model.fit(X_train, y_train)

4. 模型评估:
# 在测试集上进行预测
y_pred = model.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print(f'Accuracy: {accuracy}')

# 打印分类报告
print(classification_report(y_test, y_pred))

# 打印混淆矩阵
conf_matrix = confusion_matrix(y_test, y_pred)
print('Confusion Matrix:')
print(conf_matrix)

这个示例中,RandomForestClassifier 的 n_estimators 参数表示构建的决策树数量。你可以根据需要调整其他参数,例如树的深度、最小拆分样本数等。随机森林通常对超参数的选择不敏感,因此默认参数通常能够产生不错的结果。


转载请注明出处:http://www.pingtaimeng.com/article/detail/12004/AI人工智能