Регрессия с использованием MXNet - PullRequest
0 голосов
/ 16 апреля 2020

У меня есть модель регрессии, основанная на различных независимых функциях, которые в конечном итоге предсказывают значение с помощью пользовательской функции потерь. Несколько похоже на ссылку ниже.

https://www.evergreeninnovations.co/blog-quantile-loss-function-for-machine-learning/

Текущая модель построена с использованием библиотеки Tensorflow, но теперь я хочу использовать M XNet из-за скорости и другие преимущества, которые он обеспечивает. Как написать похожий лог c в M XNet с пользовательской функцией потерь?

1 Ответ

0 голосов
/ 17 апреля 2020

Простая регрессия с потерей L2 показана в 2 известных уроках - вы можете просто выбрать любой из них и настроить потерю:

...