REGMixin

class fl_sim.models.REGMixin

Bases: object

Mixin for regressors.

predict(input: Tensor) → ndarray

Predict the regression target.

Parameters:

input (torch.Tensor) – The input data.

Returns:

output – The predicted regression target.

Return type:

numpy.ndarray