RegressionModel#

class RegressionModel[source]#

Bases: Module

__init__(backbone: torch.nn.Module, out_features: int, pos_out_inds: Optional[Sequence[int]] = None, prob_out_inds: Optional[Sequence[int]] = None, **kwargs)[source]#
Parameters

Methods

__init__(backbone, out_features[, ...])

forward(x)

__contains__(key: str) → bool#
Parameters

key (str) –

Return type

bool

__getitem__(key: Any) → _MockObject#
Parameters

key (Any) –

Return type

_MockObject

__init__(backbone: torch.nn.Module, out_features: int, pos_out_inds: Optional[Sequence[int]] = None, prob_out_inds: Optional[Sequence[int]] = None, **kwargs)[source]#
Parameters
static __new__(cls, *args: Any, **kwargs: Any) → Any#
Parameters

  • args (Any) –

  • kwargs (Any) –

Return type

Any

forward(x: torch.Tensor) → torch.Tensor[source]#
Parameters

x (torch.Tensor) –

Return type

torch.Tensor