модуль 'trax.supervised' не имеет атрибута 'inputs'

Question

Я работаю над проектом с использованием пакета TRAX от Google. Я обнаружил следующее сообщение об ошибке при запуске фрагмента кода в записной книжке Jupyter.

      module 'trax.supervised' has no attribute 'inputs'

Есть ли способ обойти эту проблему? У меня trax версии 1.2.3.

Спасибо!

Answer 1

Согласно документации trax Вызов должен быть trax.data, а не trax. Контролируемый trax.data.inputs.add_loss_weights(generator, id_to_mask = None)

https://trax-ml.readthedocs.io/en/latest/trax.data.html