Как изменить RandomForestClassifier в Python scikit-learn - PullRequest
Новая
       6

Как изменить RandomForestClassifier в Python scikit-learn

0 голосов
/ 21 ноября 2018

Как я могу изменить RandomForestClassifier в scikit-learn, чтобы вместо удержания в каждом терминальном узле распределение классов.Я просматриваю все записи (в каждом терминальном узле) и запускаю формулу, такую ​​как linear regression? 

# Import the random forest package
from sklearn.ensemble import RandomForestClassifier
# Create the random forest object which will include all the parameters
# for the fit
forest = RandomForestClassifier(n_estimators = 1)
#Take the same decision trees and run it on the test data
output = forest.predict(test_data)
...