def test_create_label_tensors(self):
tensorizer = LabelTensorizer(column="label")
init = tensorizer.initialize()
init.send(None) # kick
for row in self.data.train:
init.send(row)
init.close()
rows = [
{
"label": types.Label("weather/find")
},
{
"label": types.Label("alarm/set_alarm")
},
{
"label": types.Label("non/existent")
},
]
tensors = (tensorizer.numberize(row) for row in rows)
tensor = next(tensors)
self.assertEqual(6, tensor)
tensor = next(tensors)
self.assertEqual(1, tensor)
with self.assertRaises(Exception):
tensor = next(tensors)
def test_create_label_tensors(self):
tensorizer = LabelTensorizer(label_column="label")
self._initialize_tensorizer(tensorizer)
rows = [
{"label": "weather/find"},
{"label": "alarm/set_alarm"},
{"label": "non/existent"},
]
tensors = (tensorizer.numberize(row) for row in rows)
tensor = next(tensors)
self.assertEqual(6, tensor)
tensor = next(tensors)
self.assertEqual(1, tensor)
with self.assertRaises(Exception):
tensor = next(tensors)
