輸出數量可變的Python Scikit決策樹

Question

我正在尋找使用Python SciKit庫設置多輸出決策樹的方法。 但是，我面臨的問題是這不是一個簡單的“ n_outputs”分類。 一些樣本將有3個輸出，一些有4個，有5個。我不確定將其傳達給庫的最佳方法是什么。

我正在考慮使用最大數量的輸出並具有“無輸出”分類。 因此，如果我訓練一個將每個樣本強制轉換為5個輸出的集合，那么原來只有3個分類的任何樣本都將通過添加“無輸出”分類而更改為5個。

您認為這樣行得通嗎？ 還有其他方法來制作輸出數量可變的多輸出決策樹嗎？

Answer 1

聽起來您正在嘗試進行多標簽分類，而不是多輸出分類。 通過提供一個指示符向量可以很容易地實現多標簽，該指示符向量針對每個樣本和每個類別，說明它們是否在類別中，因此您將得到一個大小為二進制的數組（非類別為0，類別為1） （n_samples，n_classes）。

查看多標簽文檔 ，看看是否適合您的用例。