def _get_param(self):
if self.need_cv:
param_protobuf_obj = lr_model_param_pb2.LRModelParam()
return param_protobuf_obj
self.header = self.header if self.header else []
LOGGER.debug("In get_param, self.need_one_vs_rest: {}".format(
self.need_one_vs_rest))
if self.need_one_vs_rest:
one_vs_rest_result = self.one_vs_rest_obj.save(
lr_model_param_pb2.SingleModel)
single_result = {
'header': self.header,
'need_one_vs_rest': True,
"best_iteration": -1
}
else:
one_vs_rest_result = None
single_result = self.get_single_model_param()
single_result['need_one_vs_rest'] = False
single_result['one_vs_rest_result'] = one_vs_rest_result
param_protobuf_obj = lr_model_param_pb2.LRModelParam(**single_result)
return param_protobuf_obj
def _get_param(self):
header = self.header
LOGGER.debug("In get_param, header: {}".format(header))
if header is None:
param_protobuf_obj = lr_model_param_pb2.LRModelParam()
return param_protobuf_obj
if self.need_one_vs_rest:
# one_vs_rest_class = list(map(str, self.one_vs_rest_obj.classes))
one_vs_rest_result = self.one_vs_rest_obj.save(
lr_model_param_pb2.SingleModel)
single_result = {
'header': header,
'need_one_vs_rest': True,
"best_iteration": -1
}
else:
one_vs_rest_result = None
single_result = self.get_single_model_param()
single_result['need_one_vs_rest'] = False
single_result['one_vs_rest_result'] = one_vs_rest_result
# LOGGER.debug("in _get_param, single_result: {}".format(single_result))
param_protobuf_obj = lr_model_param_pb2.LRModelParam(**single_result)
return param_protobuf_obj
def _get_param(self):
header = self.header
if header is None:
param_protobuf_obj = lr_model_param_pb2.LRModelParam()
return param_protobuf_obj
result = self._get_model_param()
param_protobuf_obj = lr_model_param_pb2.LRModelParam(**result)
return param_protobuf_obj
def _get_param(self):
header = self.header
LOGGER.debug("In get_param, header: {}".format(header))
if header is None:
param_protobuf_obj = lr_model_param_pb2.LRModelParam()
return param_protobuf_obj
if self.need_one_vs_rest:
result = self._get_model_param_ovr()
param_protobuf_obj = lr_model_param_pb2.LRModelParam(**result)
else:
result = self._get_model_param()
param_protobuf_obj = lr_model_param_pb2.LRModelParam(**result)
LOGGER.debug("in _get_param, result: {}".format(result))
return param_protobuf_obj
def _get_param(self):
if self.need_cv:
param_protobuf_obj = lr_model_param_pb2.LRModelParam()
return param_protobuf_obj
if self.need_one_vs_rest:
one_vs_rest_result = self.one_vs_rest_obj.save(
lr_model_param_pb2.SingleModel)
single_result = {
'header': self.header,
'need_one_vs_rest': True,
"best_iteration": -1
}
else:
one_vs_rest_result = None
single_result = self.get_single_model_param()
single_result['need_one_vs_rest'] = False
single_result['one_vs_rest_result'] = one_vs_rest_result
LOGGER.debug(f"saved_model: {single_result}")
param_protobuf_obj = lr_model_param_pb2.LRModelParam(**single_result)
return param_protobuf_obj
def _get_param(self):
header = self.header
weight_dict = {}
intercept = 0
if not self.use_encrypt and not self.component_properties.need_cv:
lr_vars = self.model_weights.coef_
for idx, header_name in enumerate(header):
coef_i = lr_vars[idx]
weight_dict[header_name] = coef_i
intercept = self.model_weights.intercept_
param_protobuf_obj = lr_model_param_pb2.LRModelParam(iters=self.n_iter_,
loss_history=self.loss_history,
is_converged=self.is_converged,
weight=weight_dict,
intercept=intercept,
header=header)
return param_protobuf_obj
def _get_param(self):
header = self.header
weight_dict = {}
intercept = 0
if not self.use_encrypt:
lr_vars = self.model_weights.coef_
for idx, header_name in enumerate(header):
coef_i = lr_vars[idx]
weight_dict[header_name] = coef_i
intercept = self.model_weights.intercept_
param_protobuf_obj = lr_model_param_pb2.LRModelParam(iters=self.n_iter_,
loss_history=self.loss_history,
is_converged=self.is_converged,
weight=weight_dict,
intercept=intercept,
header=header)
from google.protobuf import json_format
json_result = json_format.MessageToJson(param_protobuf_obj)
LOGGER.debug("json_result: {}".format(json_result))
return param_protobuf_obj
