Как мне настроить параметры? - PullRequest
0 голосов
/ 29 ноября 2018

У меня есть модель дерева решений, написанная с кодами ниже, могу ли я узнать, как я могу настроить параметры, чтобы улучшить результат модели?Могу ли я использовать что-то вроде GridsearchCV или что-то еще?Обучающие данные - около 5000+, в то время как тестовые данные - около 1000+ с более чем 20 функциями.

clf_entropy = DecisionTreeClassifier (критерий = "энтропия", 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 («Точность есть», точность_score (Y_test, Y_pred_en) * 100)

1 Ответ

0 голосов
/ 30 мая 2019

Похоже, вы должны использовать разные алгоритмы, такие как BaggingClassifier или RandomForestClassifier вместо поиска случайных параметров модели дерева в DecisionTreeClassifier.Плз гугл для заявленных моделей

...