def test_num(self):
x = Classifier(CONFIG)
self.assertEqual(
2, x.train([
('Y', Datum({'x': 1})),
('N', Datum({'x': -1})),
]))
def _test_classify(x):
y = x.classify([Datum({'x': 1}), Datum({'x': -1})])
self.assertEqual(['Y', 'N'], [
list(sorted(z, key=lambda x: x.score, reverse=True))[0].label
for z in y
])
self.assertEqual(x.get_labels(), {'N': 1, 'Y': 1})
_test_classify(x)
model = x.save_bytes()
x.clear()
self.assertEqual({}, x.get_labels())
x.set_label('Y')
x.set_label('N')
self.assertEqual({'N': 0, 'Y': 0}, x.get_labels())
x.delete_label(u'Y')
self.assertEqual({'N': 0}, x.get_labels())
x = Classifier(CONFIG)
x.load_bytes(model)
_test_classify(x)
self.assertEqual(CONFIG, json.loads(x.get_config()))
def test_num(self):
x = Classifier(CONFIG)
self.assertEqual(2, x.train([
('Y', Datum({'x': 1})),
('N', Datum({'x': -1})),
]))
def _test_classify(x):
y = x.classify([
Datum({'x': 1}),
Datum({'x': -1})
])
self.assertEqual(['Y', 'N'], [list(sorted(
z, key=lambda x:x.score, reverse=True))[0].label for z in y])
self.assertEqual(x.get_labels(), {'N': 1, 'Y': 1})
_test_classify(x)
model = x.save_bytes()
x.clear()
self.assertEqual({}, x.get_labels())
x.set_label('Y')
x.set_label('N')
self.assertEqual({'N': 0, 'Y': 0}, x.get_labels())
x.delete_label(u'Y')
self.assertEqual({'N': 0}, x.get_labels())
x = Classifier(CONFIG)
x.load_bytes(model)
_test_classify(x)
self.assertEqual(CONFIG, json.loads(x.get_config()))
def test_num(self):
x = Classifier(CONFIG)
self.assertEqual(
2, x.train([
('Y', Datum({'x': 1})),
('N', Datum({'x': -1})),
]))
def _test_classify(x):
y = x.classify([Datum({'x': 1}), Datum({'x': -1})])
self.assertEqual(['Y', 'N'], [
list(sorted(z, key=lambda x: x.score, reverse=True))[0].label
for z in y
])
self.assertEqual(x.get_labels(), {'N': 1, 'Y': 1})
_test_classify(x)
model = x.save_bytes()
self.assertTrue(x.clear())
self.assertEqual({}, x.get_labels())
x.set_label('Y')
x.set_label('N')
self.assertEqual({'N': 0, 'Y': 0}, x.get_labels())
x.delete_label(u'Y')
self.assertEqual({'N': 0}, x.get_labels())
x = Classifier(CONFIG)
x.load_bytes(model)
_test_classify(x)
self.assertEqual(CONFIG, json.loads(x.get_config()))
if sys.version_info[0] == 3:
x = pickle.loads(pickle.dumps(x))
_test_classify(x)
self.assertEqual(CONFIG, json.loads(x.get_config()))
st = x.get_status()
self.assertTrue(isinstance(st, dict))
self.assertEqual(len(st), 1)
self.assertEqual(list(st.keys())[0], 'embedded')
self.assertTrue(isinstance(st['embedded'], dict))
