Module corgi::cost

Cost functions compute the loss given a target, and are used for the backward pass.

Functions

cross_entropy

Creates a cross-entropy loss closure.

mse

Creates a mean square error loss closure.

Type Definitions

CostFunction

A cost function, which computes the loss given a target. The cost function takes in the output as the first argument, and the target as the second.