GaleShapleyMatcher¶
- class GaleShapleyMatcher(segments: List[etna.transforms.feature_selection.gale_shapley.SegmentGaleShapley], features: List[etna.transforms.feature_selection.gale_shapley.FeatureGaleShapley])[source]¶
Bases:
etna.core.mixins.BaseMixin
Class for handling Gale-Shapley matching algo.
Init GaleShapleyMatcher.
- Parameters
segments (List[etna.transforms.feature_selection.gale_shapley.SegmentGaleShapley]) – list of segments to build matches
features (List[etna.transforms.feature_selection.gale_shapley.FeatureGaleShapley]) – list of features to build matches
- Inherited-members
Methods
break_match
(segment, feature)Break match between segment and feature.
match
(segment, feature)Build match between segment and feature.
set_params
(**params)Return new object instance with modified parameters.
to_dict
()Collect all information about etna object in dict.
- static break_match(segment: etna.transforms.feature_selection.gale_shapley.SegmentGaleShapley, feature: etna.transforms.feature_selection.gale_shapley.FeatureGaleShapley)[source]¶
Break match between segment and feature.
- Parameters
segment (etna.transforms.feature_selection.gale_shapley.SegmentGaleShapley) – segment to break match
feature (etna.transforms.feature_selection.gale_shapley.FeatureGaleShapley) – feature to break match
- static match(segment: etna.transforms.feature_selection.gale_shapley.SegmentGaleShapley, feature: etna.transforms.feature_selection.gale_shapley.FeatureGaleShapley)[source]¶
Build match between segment and feature.
- Parameters
segment (etna.transforms.feature_selection.gale_shapley.SegmentGaleShapley) – segment to match
feature (etna.transforms.feature_selection.gale_shapley.FeatureGaleShapley) – feature to match