docs(network): add docstring for MonomialNN
@@ -70,6 +70,13 @@ class Resnet(nn.Module):
class MonomialNN(nn.Module):
+ """A network that expands its input to a given list of monomials
+ Its output shape will be (n_samples, n_input_units * n_degrees)
+ :param n_input_units: number of input features
+ :type n_input_units: int
+ :param degrees: max degree to be included, or a list of degrees that will be used
+ :type degrees: int or list[int] or tuple[int]
+ """
def __init__(self, n_input_units, degrees=1):
super(MonomialNN, self).__init__()
