随机森林

最新推荐文章于 2023-02-14 18:05:36 发布

lcp0633

最新推荐文章于 2023-02-14 18:05:36 发布

阅读量272

点赞数

分类专栏：机器学习sklearn 文章标签：机器学习

本文链接：https://blog.csdn.net/lcp0633/article/details/106032456

版权

机器学习sklearn 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_wine
from sklearn.model_selection import train_test_split

# 1、加载数据集
x, y = load_wine(return_X_y=True)
print(x.shape)
print(y.shape)
print(x)
print(y)

x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3)
# 2、创建模型
model_tree = DecisionTreeClassifier()
model_forest = RandomForestClassifier()
# 3、训练模型
model_tree.fit(x_train, y_train)
model_forest.fit(x_train, y_train)

# 4、测试模型
score1 = model_tree.score(x_test, y_test)
score2 = model_forest.score(x_test, y_test)

print(score1)
print(score2)