私は平均したいクラスIDと予測確率の大規模な行列を扱っています。次に、各行で確率が最も高い 3 つのクラスを返したいと考えています。
問題は、各行のクラスが異なることです。これを実装する最も効率的な方法は何ですか?
以下は、1 行だけを使用したおもちゃの例です。
a = [11, 12, 13]
a_probs = [0.2, 0.1, 0.02]
b = [8, 11, 15]
b_probs = [0.05, 0.4, 0.12]
したがって、この例では、クラス 11 のみが両方の行列で発生します。したがって、各クラスの平均確率は次のようになります。
[8, 11, 12, 13, 15] (0.05+0)/2 (0.2+0.4)/2 0.1+0/2 0.02+0/2 0.12+0/2
私の現在の方法は非常に遅いです.すべての行列で1行のクラスを連結し、一意であり、各クラスの確率を見つけて合計し、平均します.