TF JS Категориальная кроссентропия с использованием from_logits

Question

Я новичок в ML, и какое-то время боролся. Я конвертирую свой старый python скрипт в javascript версию tfjs. Однако я не могу найти одну строку в python, которая равна tf js.

Im конвертирую python script:

model.compile(optimizer='adam',
  loss=tf.losses.CategoricalCrossentropy(from_logits=True),
  metrics=['accuracy'])

в javascript tf js script :

model.compile({
  optimizer: tf.train.adam(),
  loss: tf.losses.softmaxCrossEntropy,
  metrics: ['accuracy'],
});

Я не смог найти tf js версию tf.losses.CategoricalCrossentropy(from_logits=True), особенно переменную from_logits.

Любая помощь будет принята с благодарностью. Заранее спасибо.