Добрый день! Написал кастомный Focal loss на c++ для CatboostClassifier,

Question

Добрый день! Написал кастомный Focal loss на c++ для CatboostClassifier,

все работает, и есть два вопроса:

1. Могу сделать ПР с ним на гитхабе, но лосс сейчас а) не покрыт тестами б)не проверялся на ГПУ(подозреваю, его нужно переложить на куду). Интересен такой ПР создателям? Или интересен только если все допилить до идеального вида(могу в целом с вашими ревью/подсказками)

2. Мне нужно было протащить в лоссы в целом пенальти в виде значения одной из фич, попадающих на вход катбуста. В lightgbm все фичи доступны в лоссах из коробки, т.к. передаются вместе с таргетом, в катбусте - нет.
Я заколхозил это через присваивание значения фичи параметру weights python-Pool, которые передаются в лосс, и изменение логики его работы, но если вы подскажете более логичный способ, я могу это сделать и так же оформить ПР. Боюсь, сам через все это нагромождение классов я нормально фичу в лоссы не протяну.

#catboost #database #russian

0

13.08.2021

1 ответов

32 просмотра

Stanislav Kirillov · Accepted Answer

Stanislav Kirillov

Конечно, принимаем PR:) по поводу GPU и тестов - можно в процессе ревью все это починить

0

13.08.2021

16 похожих чатов

Добрый день! Написал кастомный Focal loss на c++ для CatboostClassifier,

1 ответов

Похожие вопросы