python - 随机森林分类的​​不确定性

Question

我正在使用 Python 中的随机森林分类算法对卫星图像进行分类。我想在像素级别映射每个类别的分类不确定性。

可能吗？

到目前为止，我在互联网上搜索并找到了这个链接https://www.sciencedirect.com/science/article/pii/S0303243412001195但这里提到的计算不确定性的方法仅适用于二元分类。

我非常感谢这方面的任何指导。

Answer 1

根据您提到的出版物:

The classification uncertainty of a pixel u is characterized by the probability vector (pu) obtained by the probabilistic classifier. This vector contains the probability p(i) of being classified into class i (pu = (p(1), p(2), …, p(c)), with c the total number of land cover classes

我建议使用 sklearn.ensemble.RandomForestClassifier 的 predict_proba 函数，它返回具有上述概率的形状数组(n_samples，n_classes)。

希望有帮助。