def _get_param(self):
pb_dict = {}
for col_name, pair_obj in self.col_maps.items():
values = [str(x) for x in pair_obj.values]
value_dict_obj = onehot_param_pb2.ColsMap(values=values,
transformed_headers=pair_obj.transformed_headers)
pb_dict[col_name] = value_dict_obj
result_obj = onehot_param_pb2.OneHotParam(col_map=pb_dict,
result_header=self.inner_param.result_header)
return result_obj
def _get_param(self):
pb_dict = {}
LOGGER.debug("in save model, col_maps: {}".format(self.col_maps))
for col_name, value_dict in self.col_maps.items():
values = list(value_dict.keys())
values = sorted(values)
LOGGER.debug("In _get_param, values: {}".format(values))
data_type = type(values[0]).__name__
encoded_variables = []
for v in values:
encoded_variables.append(value_dict[v])
values = [str(x) for x in values]
value_dict_obj = onehot_param_pb2.ColsMap(
values=values,
encoded_variables=encoded_variables,
data_type=data_type)
pb_dict[col_name] = value_dict_obj
result_obj = onehot_param_pb2.OneHotParam(col_map=pb_dict)
return result_obj
