def test_schedule(self): print "test_schedule" tasks = [Task(0,1,1), Task(0,2,1), Task(1,4,1), Task(4,5,1), Task(2,6,1), Task(5,7,1)] set_correct = [Task(0,1,1), Task(1,4,1), Task(4,5,1), Task(5,7,1)] max_w, set_tasks = wis.schedule(tasks) print set_correct print set_tasks self.assertEqual(set_correct, set_tasks)
def argmax(w, e, loss=0): tasks = [] i = 0 for r in e.x: l = 0 if loss > 0 and not find(r, e.y): l = loss coref = w * np.array(r.coref.feat) t = wis.Task(r.quote.start, r.quote.end, np.sum(coref) + l, i) tasks.append(t) i += 1 wmax, set_tasks = wis.schedule(tasks) result = convertTasks(set_tasks, e.x) return result