У меня есть датафрейм в форме, где строки указывают идентификаторы, а столбцы указывают количество классов.
Скажем, у меня есть 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])
Как лучше всего добавить этот тензор в конкретную строку, для которой я вычисляю эти вероятности. Учитывая, что я хочу добавить его в dataframe.
Кроме того, как я могу ограничить конечные числа после десятичной дроби в тензоре?