def get_state(self):
state = super().get_state()
state.update({
'encoder': encoder.serialize(self.label_encoder),
'encoder_state': self.label_encoder.get_state() if self.label_encoder
else None})
return state
def get_state(self):
state = super().get_state()
state.update({
'one_hot_encoder':
encoder.serialize(self._one_hot_encoder),
'one_hot_encoder_state':
self._one_hot_encoder.get_state()
if self._one_hot_encoder else None,
'num_classes':
self.num_classes
})
return state
def get_state(self):
state = super().get_state()
label_encoders = {
key: encoder.serialize(label_encoder)
for key, label_encoder in self.label_encoders.items()
}
label_encoders_state = {
key: label_encoder.get_state()
for key, label_encoder in self.label_encoders.items()
}
state.update({
'shape':
self.shape,
'num_rows':
self.num_rows,
'categorical_col':
self.categorical_col,
'numerical_col':
self.numerical_col,
'label_encoders':
utils.to_type_key(label_encoders, str),
'label_encoders_state':
utils.to_type_key(label_encoders_state, str),
'value_counters':
utils.to_type_key(self.value_counters, str),
'categorical_categorical':
utils.to_type_key(self.categorical_categorical, str),
'numerical_categorical':
utils.to_type_key(self.numerical_categorical, str),
'count_frequency':
utils.to_type_key(self.count_frequency, str),
'high_level1_col':
self.high_level1_col,
'high_level2_col':
self.high_level2_col,
'high_level_cat_cat':
utils.to_type_key(self.high_level_cat_cat, str),
'high_level_num_cat':
utils.to_type_key(self.high_level_num_cat, str),
'column_names':
self.column_names,
'column_types':
utils.to_type_key(self.column_types, str),
'num_columns':
self.num_columns,
})
return state
