# flambe.model.logistic_regression¶

## Module Contents¶

class flambe.model.logistic_regression.LogisticRegression(input_size: int)[source]

Logistic regression model given an input vector v the forward calculation is sigmoid(Wv+b), where W is a weight vector and b a bias term. The result is then passed to a sigmoid function, which maps it as a real number in [0,1]. This is typically interpreted in classification settings as the probability of belonging to a given class.

input_size

Dimension (number of features) of the input vector.

Type: int
forward(self, data: Tensor, target: Optional[Tensor] = None)[source]

Forward pass that encodes data :param data: input data to encode :type data: Tensor :param target: target value, will be casted to a float tensor. :type target: Optional[Tensor]