Skip to content

Optimize categorical crossentropy gradient update #91

@NicolasHug

Description

@NicolasHug

The gradient and hessian update of the categorical crossentropy loss computes p_k k times, but it only needs to compute it once (see scikit-learn/scikit-learn@9e68984 which led to serious improvement in fit time).

Metadata

Metadata

Assignees

Labels

perfComputational performance issue or improvement

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions