flambe.model
¶
Submodules¶
Package Contents¶
-
class
flambe.model.
LogisticRegression
(input_size: int)[source]¶ Bases:
flambe.nn.module.Module
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)¶ 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]
-