File tree Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Original file line number Diff line number Diff line change @@ -109,14 +109,21 @@ def __init__( self,
109
109
def mutate (self , rng = None ):
110
110
rng = np .random .default_rng (rng )
111
111
#get list of names not including the current one
112
- names = [name for name in self .names_list if name != self .selected_subset_name ]
113
- self .selected_subset_name = rng .choice (names )
114
- self .sel_subset = self .subset_dict [self .selected_subset_name ]
112
+ if len (self .names_list )> 1 :
113
+ names = [name for name in self .names_list if name != self .selected_subset_name ]
114
+ self .selected_subset_name = rng .choice (names )
115
+ self .sel_subset = self .subset_dict [self .selected_subset_name ]
116
+ return True
117
+ else :
118
+ return False
115
119
116
120
117
121
def crossover (self , other , rng = None ):
122
+ if self .selected_subset_name == other .selected_subset_name :
123
+ return False
118
124
self .selected_subset_name = other .selected_subset_name
119
125
self .sel_subset = other .sel_subset
126
+ return True
120
127
121
128
def export_pipeline (self , ** kwargs ):
122
129
return FeatureSetSelector (sel_subset = self .sel_subset , name = self .selected_subset_name )
You can’t perform that action at this time.
0 commit comments