wip for levi's idea

Author: sumedhpendurkar

src/curriculum/curriculum.py

@@ -28,6 +28,7 @@ def __init__(self, num_states, model_name, state_generator, test_set_path,\
         self._time = [0]
         self._solution_quality = [0]
         self._solution_expansions = [0]
+        self._traj = []
 
 
         self._log_folder = 'training_logs/'
@@ -73,6 +74,7 @@ def solve(self, states, planner, nn_model, budget, memory, update:bool):
                 total_expanded += result[2]
                 total_generated += result[3]
                 puzzle_name = result[4]
+                self._traj.append(trajectory)
 
                 if has_found_solution:
                     #print(trajectory.get_solution_costs())
@@ -103,6 +105,11 @@ def solve(self, states, planner, nn_model, budget, memory, update:bool):
             batch_problems.clear()
         return (sum_sol_cost, number_solved, total_expanded, total_generated, sol_costs, sol_expansions)
 
+    def get_traj(self):
+        traj = self._traj
+        self._traj = []
+        return traj
+
     @abstractmethod
     def learn_online(self, planner, nn_model):
         pass

src/curriculum/lcb_curriculum.py

@@ -73,9 +73,12 @@ def learn_online(self, planner, nn_model):
                 states[i], difficulty = self.generate_state(nn_model, base_state, new_budget)
                 difficulties.append(difficulty)
                 #print(states[i], difficulty)
+
+            self._traj = []
             _, number_solved, total_expanded, total_generated, sol_costs, sol_expansions = self.solve(states,
                         planner=planner, nn_model=nn_model, budget=budget, memory=memory, update=True)
 
+            trajs = self.get_traj()
             staters_per_itr = states
             expansions_per_tr = sol_expansions
             end = time.time()