GPT机器人更新:可正常显示数学公式;添加了raw模式(回复原始HTML代码)
『回复列表(30|隐藏机器人聊天)』
[新会话] @帝国刺客,以下是一个简单的决策树的 Python 实现代码,使用 scikit-learn 库:
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
# 加载鸢尾花数据集
iris = load_iris()
X = iris.data
y = iris.target
# 将数据集划分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建决策树分类器
clf = DecisionTreeClassifier(random_state=42)
# 使用训练数据拟合分类器
clf.fit(X_train, y_train)
# 使用测试数据进行预测
y_pred = clf.predict(X_test)
# 输出预测结果
print("Predicted class:", y_pred[0])
print("True class:", y_test[0])
print("Accuracy:", clf.score(X_test, y_test))
在这个例子中,我们使用了 scikit-learn 库中的 DecisionTreeClassifier
类来创建决策树分类器。我们使用 load_iris()
函数加载鸢尾花数据集,并将数据集划分为训练集和测试集。然后,我们创建了一个决策树分类器,并使用训练数据对其进行拟合。最后,我们使用测试数据进行预测,并输出预测结果和准确率。
请注意,这只是一个简单的决策树实现示例,实际应用中可能需要进行更多的数据预处理、特征工程和模型调参等步骤。