diff --git a/dtree_synth_agbot.py b/dtree_synth_agbot.py index c21f1cca083b5486462de044a490c8d70abc8941..92b66d193b12d58e7bbb8a84391934fbdc90af31 100644 --- a/dtree_synth_agbot.py +++ b/dtree_synth_agbot.py @@ -32,8 +32,8 @@ def load_examples_from_npz(file_name: str, teacher:AgBotTeacher, partition): data_5d_numpy =npz_data['dps_arr'] data_5d = list(map(tuple, data_5d_numpy)) # x_arr[:-1], except the last one, x_arr[1:]except the first one - print(data_5d[0]) - print(len(data_5d)) + bound_list = list(list(zip(x_arr[:-1], x_arr[1:])) for x_arr in partition) + ret = {part: [[], [], 0] for part in itertools.product(*bound_list)} print(teacher.is_positive_example(data_5d[0]))