bob.med.tb.models.logistic_regression¶

Functions

build_logistic_regression(input_size)

Build logistic regression module

Classes

LogisticRegression(input_size)

Radiological signs to Tuberculosis module

class bob.med.tb.models.logistic_regression.LogisticRegression(input_size)[source]¶

Bases: Module

Radiological signs to Tuberculosis module

forward(x)[source]¶

Parameters: x (list) – list of tensors.
Returns: tensor
Return type: torch.Tensor

bob.med.tb.models.logistic_regression.build_logistic_regression(input_size)[source]¶

Build logistic regression module

Returns: module
Return type: torch.nn.Module