Я использую scikit-learn для принятия дерева решений, и мне нужно знать количество узлов каждой функции и значения отсечения для каждого узла.Например, в дереве я хочу узнать, сколько узлов имеет переменная «size», или сколько узлов имеет переменная «location» в дереве, и каковы предельные значения в этих узлах, если это возможно.
Вот мои примерные данные обучения:
size,count,status,location,is_relevant
0.99,39.19,true,east,false
6.45,58.308,false,south,false
4.844,103.6,false,west,true
4.1,63.209,false,west,false
1.05,52.06,false,west,false
Вот мои примеры данных тестирования:
size,count,status,location,is_relevant
2.88,139.78,false,north,true
1.29,11.37,false,south,false
3.6,33.97,false,west,true
1.8,43.93,false,west,false
7.41,60.2,false,south,false
Вот код, который я использую для создания дерева:
training['location']=pd.Categorical(training.location).codes
testing['location']=pd.Categorical(testing.location).codes
y=training.loc[:,'is_relevant']
x=training.iloc[:, :-1]
from sklearn import tree
model = tree.DecisionTreeClassifier()
model=model.fit(x, y)
predictions = model.predict(testing.iloc[:, :-1])
print(model.tree_)
print(model.tree_.node_count)
Мне удалось экспортировать и визуализировать дерево, но слишком много узлов, чтобы просто пойти и сосчитать количество узлов для нужной мне функции.Есть команда для этого?
Спасибо