def create_inputs_map(values_map, input_config): inputs = {} for input_name, input_config_item in input_config.items(): if util.is_str(input_config_item): inputs[input_name] = values_map[input_config_item] elif util.is_int(input_config_item): inputs[input_name] = values_map[input_config_item] elif util.is_list(input_config_item): inputs[input_name] = [values_map[f] for f in input_config_item] else: raise CortexException("invalid column inputs") return inputs
def cast_compound_type(value, type_str): allowed_types = type_str.split("|") if consts.VALUE_TYPE_INT in allowed_types: if util.is_int(value): return value if consts.VALUE_TYPE_FLOAT in allowed_types: if util.is_int(value): return float(value) if util.is_float(value): return value if consts.VALUE_TYPE_STRING in allowed_types: if util.is_str(value): return value if consts.VALUE_TYPE_BOOL in allowed_types: if util.is_bool(value): return value raise UserException( "unsupported input type (expected type {}, got {})".format( util.data_type_str(type_str), util.user_obj_str(value) ) )
def create_inputs_from_features_map(features_values_map, feature_input_config): inputs = {} for input_name, input_config_item in feature_input_config.items(): if util.is_str(input_config_item): inputs[input_name] = features_values_map[input_config_item] elif util.is_int(input_config_item): inputs[input_name] = features_values_map[input_config_item] elif util.is_list(input_config_item): inputs[input_name] = [ features_values_map[f] for f in input_config_item ] else: raise CortexException("invalid feature inputs") return inputs