BinaryClassifier

class deeplay.applications.classification.binary.BinaryClassifier(*args, **kwargs)

Bases: Application

Methods Summary

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

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

metrics_preprocess(y_hat, y: Tensor)