ánh xạ scikit-learning DecisionTreeClassifier.tree_.value để dự đoán lớp

Question

Tôi đang sử dụng một DecissionTreeClassifier scikit-learning trên một tập dữ liệu 3 lớp. Sau khi tôi phù hợp với trình phân loại, tôi truy cập tất cả các nút lá trên thuộc tính tree_ để nhận được số lượng các cá thể kết thúc trong một nút nhất định cho mỗi lớp.

clf = tree.DecisionTreeClassifier(max_depth=5) 
clf.fit(X, y) 
# lets assume there is a leaf node with id 5 
print clf.tree_.value[5] 

này sẽ in ra:

>>> array([[ 0., 1., 68.]]) 

nhưng ... làm thế nào để tôi biết được vị trí trong mảng đó thuộc mà lớp học? Bộ phân loại có một thuộc tính classes_ đó cũng là một danh sách

>>> clf.classes_ 
array(['CLASS_1', 'CLASS_2', 'CLASS_3'], dtype=object) 

Có lẽ chỉ số 1 về mảng giá trị phù hợp với lớp trên chỉ số 1 của mảng lớp và vân vân?

Answer 1

Được hỏi về điều này trong danh sách gửi thư scikit-learm và dự đoán của tôi là đúng. Chỉ ra chỉ số 1 trên mảng giá trị khớp với lớp trên chỉ mục 1 của mảng lớp và như vậy trên

Answer 2

Không, nó không phải là clf.classes_ nhưng clf.tree_.feature chứa chỉ mục cột của X. Và, nếu X là Pandas DataFrame, X.columns chứa tên. Bạn có thể tìm thêm thông tin chi tiết trong a similar question.