在数据框中将张量添加为一行

时间:2018-10-30 17:58:54

标签: dataframe pytorch tensor

我有一个数据框,其格式为:行表示ID,列表示类数。 假设我有10个课程,则列名称将为:

‘class1’, ‘class2’, . . . ‘class10’

对于每一行,我都在预测属于各种类别的概率。 假设有9个类别,这些是存储在张量中的计算出的概率:

tensor([1.3953e-05, 3.3629e-01, 2.9408e-01, 3.5847e-01, 1.8039e-05, 9.7759e-06, 1.1096e-02, 7.1202e-06, 1.9519e-05])

将这个张量添加到我要为其计算这些概率的特定行中的最佳方法是什么。鉴于我想将其添加到数据框。 另外,如何限制张量中小数点后的数字位数?

0 个答案:

没有答案