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
