View source on GitHub
|
Specifies neighbor attributes for graph regularization.
nsl.configs.GraphNeighborConfig(
prefix=attr_dict['prefix'].default,
weight_suffix=attr_dict['weight_suffix'].default,
max_neighbors=attr_dict['max_neighbors'].default
)
Attributes | |
|---|---|
prefix
|
The prefix in feature names that identifies neighbor-specific features. Defaults to 'NLnbr'. |
weight_suffix
|
The suffix in feature names that identifies the neighbor
weight value. Defaults to '_weight'. Note that neighbor weight features
will have prefix as a prefix and weight_suffix as a suffix. For
example, based on the default values of prefix and weight_suffix, a
valid neighbor weight feature is 'NL_nbr_0_weight', where 0 corresponds to
the first neighbor of the sample.
|
max_neighbors
|
The maximum number of neighbors to be used for graph regularization. Defaults to 0, which disables graph regularization. Note that this value has to be less than or equal to the actual number of neighbors in each sample. |
Methods
__eq__
__eq__(
other
)
Method generated by attrs for class GraphNeighborConfig.
__ge__
__ge__(
other
)
Method generated by attrs for class GraphNeighborConfig.
__gt__
__gt__(
other
)
Method generated by attrs for class GraphNeighborConfig.
__le__
__le__(
other
)
Method generated by attrs for class GraphNeighborConfig.
__lt__
__lt__(
other
)
Method generated by attrs for class GraphNeighborConfig.
__ne__
__ne__(
other
)
Method generated by attrs for class GraphNeighborConfig.