Блог

у меня есть несколько столбцов функций и столбец результатов, в котором я хочу предсказать, произойдет что-то или нет. итак, я тренирую свою модель и, наконец, делаю

 const predictions = model.predict(xTest).argMax(-1);
  

это возвращает тензор и при получении данных с:

 predictions.dataSync ()
  

я получаю такие значения, как [0, 1, 1, 1, 0, …]
есть ли какой-нибудь способ получить вероятности, как в python? [0.121, 0.421, 0.8621, …]

Я нашел только один результат: https://groups.google.com/a/tensorflow.org/g/tfjs/c/TvcB69MUj_I?pli=1

это все еще так? в javascript нет вероятностей?

tf.argMax возвращает индексы максимального значения вдоль оси. Если вы скорее хотите иметь максимальное значение, которое вы могли бы использовать tf.max вместо этого

 const x = tf.tensor2d([[1, 2, 3],[ 4, 8, 4]]);

x.max(-1).print()  // [3, 8]
x.argMax(-1).print() // [2, 1]