【发布时间】:2023-01-01 04:55:44
【问题描述】:
我使用决策树算法在决策树上获得了 100% 的准确率,但在随机森林上仅获得了 75% 的准确率
我的模型有问题还是决策树最适合所提供的数据集?
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X,y,test_size = 0.3, random_state= 30)
from sklearn.preprocessing import StandardScaler
sc_X = StandardScaler()
X_train = sc_X.fit_transform(X_train)
X_test = sc_X.transform(X_test)
from sklearn.tree import DecisionTreeClassifier
classifier = DecisionTreeClassifier()
classifier = classifier.fit(X_train,y_train)
y_pred = classifier.predict(X_test)
from sklearn.metrics import confusion_matrix
cm = confusion_matrix(y_test,y_pred)
print(cm)
【问题讨论】:
-
决策树会过度拟合,而随机森林不会
标签: machine-learning decision-tree sklearn-pandas