在 scikit learn 中更改数字格式

Changing number format in scikit learn

我正在使用库 sklearn 获取 class 概率并使用 numpy.savetxt.[=16 将这些 class 概率保存在 csv 文件中=]

保存在 csv 文件中的格式是: 1.00E+00, 1.00E-01, 2.00E-01。 但是我想要 1, 0.1, 0.2 的格式。 我该怎么做?

np.savetxt 支持 fmt 参数,对于具有单精度的值,您可以传递参数 fmt=%1.1f 这应该允许概率值采用您想要的格式。

示例:

In [4]:

a = np.random.randn(5)
a
Out[4]:
array([ 0.85185988, -0.85148163,  0.5456541 , -0.26908162, -0.11920702])

np.savetxt(r'c:\data\np.dat', a, fmt='%1.1f')

产量:

0.9 -0.9 0.5 -0.3 -0.1