如何使用R中的xgboost包解釋多類分類的輸出？

Question

我是XGBOOST包的新手，我想用它來解決多類分類問題。

我現在有30個類（或標簽），我已將它們轉換為整數（0 - 29）。 然后我從數據集上的XGBOOST包運行xgboost（）方法。 構建模型后，我使用predict（）方法來預測新的測試數據。 最后，我得到了一個數字向量作為最終輸出。 順便說一句，所有數字都在0到1之間。

那么現在我該如何解釋輸出呢？ 這是概率嗎？ 但是我怎么知道應該與哪個班級相關呢？

謝謝

Answer 1

我剛拿到它！ 輸出是100,000 * 30值的矩陣。 因此，每30個數字對應一個特定實例，這意味着每個數字是該實例作為標簽的概率。

例如：

matrix [1]是Prob（instance0 == label0）

matrix [32]是Prob（instance1 == label1）

矩陣[33] =習題（INSTANCE1 == LABEL3）