def test_num_classes(self): task = TestTask() task.set_input(0, self.vowpal_example) vowpal_util.num_classes(task) self.assertEqual(int(task.outputs[0].read()), 2)
def test_convert_format(self): task = TestTask() task.set_input(0, self.libsvm_example) vowpal_util.convert_libsvm_format_to_vowpal(task) self.assertEqual(task.outputs[0].read(), self.vowpal_example)
def _process_task(self, data): task = TestTask() task.set_input(0, '\n'.join([' '.join([str(e) for e in line]) for line in data]) + '\n') task.outputs.setsize(3) rule = rules.segment_without_label_bias(self.weights) rule.fun(task) return task