我正在处理我想要平均的类 ID 和预测概率的大型矩阵。然后我想返回每行中概率最高的 3 个类。
问题是，每一行中的类各不相同。实现这一点的最有效方法是什么？
这是一个仅使用一行的玩具示例：
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
我当前的方法非常慢：在所有矩阵中连接一行的类，唯一的，定位并求和每个类的概率，平均值。