CategoricalClassifier

class deeplay.applications.classification.categorical.CategoricalClassifier(*args, **kwargs)

Bases: Application

Methods Summary

compute_loss(y_hat, y)
forward(x)	Same as torch.nn.Module.forward().

Methods Documentation

compute_loss(y_hat, y)

forward(x)

Same as torch.nn.Module.forward().

Args:

*args: Whatever you decide to pass into the forward method. **kwargs: Keyword arguments are also possible.

Return:

Your model’s output