scikit學習隨機森林的概率分佈？

Question

我有一個狀態 - >行爲對的數據集，（s，a），其中每個s定義了a的可能選擇的概率分佈，並且每個a從該概率分佈中採樣。我想爲這個數據集訓練一個分類器，而不是學習預測最大可能性，它會預測從中抽取的分佈。例如，如果您正在播放迭代搖滾剪刀，則您的狀態可能就是您之前做出的移動以及一個∈{Rock，Paper，Scissors}，其中之前的狀態會降低選擇的概率該行動再次。那麼我的數據集將是這樣的：

PreviousAction,Chosen 
Rock,Paper 
Paper,Rock 
Rock,Scissors 
Scissors,Paper 
Paper,Paper 
... 

是否可以學習在標籤概率分佈在scikit學習隨機森林？

Answer 1

是的。訓練RandomForestClassifier使用fit（它期望標籤，而不是概率分佈，作爲其參數y），然後使用predict_proba預測。