Spaces:
Sleeping
Sleeping
File size: 2,449 Bytes
f3b11f9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | import numpy as np
import pandas as pd
import configuration.config_default as cfgd
STEP_pki = 1
def encode_property_change(input_data_path, LOG=None):
property_change_encoder = {}
for property_name in cfgd.PROPERTIES:
if property_name == 'pki':
# intervals ['(3,4]', ...] 形式
# start_map_interval <number, interval> 用来找区间
intervals, start_map_interval = build_intervals(input_data_path, step=STEP_pki, LOG=LOG)
if property_name == 'pki':
property_change_encoder[property_name] = intervals, start_map_interval
return property_change_encoder
def value_in_interval(value, start_map_interval):
start_vals = sorted(list(start_map_interval.keys()))
return start_map_interval[start_vals[np.searchsorted(start_vals, value, side='right') - 1]]
def interval_to_onehot(interval, encoder):
return encoder.transform([interval]).toarray()[0]
def build_intervals(input_transformations_path, step=STEP_pki, LOG=None):
df = pd.read_csv(input_transformations_path)
# df=input_transformations_path
delta_pki = df['Delta_pki'].tolist()
min_val, max_val = min(delta_pki), max(delta_pki)
if LOG:
LOG.info("pki min and max: {}, {}".format(min_val, max_val))
start_map_interval = {}
interval_str = '({}, {}]'.format(round(-step/2, 2), round(step/2, 2))
intervals = [interval_str]
start_map_interval[-step/2] = interval_str
smallStep=step
bigStep=2
positives = step/2
while positives < 10:
if positives>2:
step=bigStep
interval_str = '({}, {}]'.format(round(positives, 2), round(positives+step, 2))
intervals.append(interval_str)
start_map_interval[positives] = interval_str
positives += step
interval_str = '({}, inf]'.format(round(positives, 2))
intervals.append(interval_str)
start_map_interval[positives] = interval_str
step=smallStep
negatives = -step/2
while negatives > min_val:
interval_str = '({}, {}]'.format(round(negatives-step, 2), round(negatives, 2))
intervals.append(interval_str)
negatives -= step
start_map_interval[negatives] = interval_str
interval_str = '(-inf, {}]'.format(round(negatives, 2))
intervals.append(interval_str)
start_map_interval[float('-inf')] = interval_str
return intervals, start_map_interval
|