在MATLAB中将概率输出转换为二进制

Question

假设我们在MATLAB R2015a中有这个矩阵：

a = [0.97 0.03
0.6 0.4
0.3 0.7];

我想将其转换为：

a1 = [1 0
1 0
0 1]

我该怎么做？

Answer 1

通过使用round（），它将所有内容舍入到最接近的整数，正如您在此处所做的那样。

a1 = round(a)

MATLAB's round() documentation

Answer 2

>> a = [0.97 0.03
0.6 0.4
0.3 0.7];

>> a > 0.5

ans =

     1     0
     1     0
     0     1