【发布时间】:2019-05-01 04:34:43
【问题描述】:
我有一个用以下代码编写的决策树模型,请问我该如何调整参数以使模型的结果更好?我可以使用 GridsearchCV 之类的东西或其他东西吗?训练数据约 5000+,测试数据约 1000+,有 20 多个特征。
clf_entropy = DecisionTreeClassifier(criterion = "entropy", random_state = 100, max_depth = 8, min_samples_leaf = 28)
clf_entropy.fit(X_train, Y_train)
Y_pred_en = clf_entropy.predict(X_test)
print("精度为", accuracy_score(Y_test, Y_pred_en) * 100)
【问题讨论】:
-
以文本形式提供代码而不是链接。
标签: python parameters decision-tree