def test_classifier_str(): x = Classifier(CLASSIFIER_CONFIG) x.train([ (u'Y', Datum({'x': u'y'})), (u'N', Datum({'x': u'n'})), ]) y = x.classify([Datum({'x': 'y'}), Datum({'x': 'n'})]) assert [ list(sorted(z, key=lambda x: x.score, reverse=True))[0].label for z in y ] == ['Y', 'N']
def test_classifier_str(): x = Classifier(CLASSIFIER_CONFIG); x.train([ (u'Y', Datum({'x': u'y'})), (u'N', Datum({'x': u'n'})), ]) y = x.classify([ Datum({'x': 'y'}), Datum({'x': 'n'}) ]) assert [list(sorted(z, key=lambda x:x.score, reverse=True))[0].label for z in y] == ['Y', 'N']
def test_classifier_num(): x = Classifier(CLASSIFIER_CONFIG) x.train([ (u'Y', Datum({'x': 1})), (u'N', Datum({'x': -1})), ]) y = x.classify([Datum({'x': 1}), Datum({'x': -1})]) assert [ list(sorted(z, key=lambda x: x.score, reverse=True))[0].label for z in y ] == ['Y', 'N'] assert sorted(x.get_labels()) == [u'N', u'Y'] model = x.dump() x.clear() assert len(x.get_labels()) == 0 x.set_label(u'Y') x.set_label(u'N') assert sorted(x.get_labels()) == [u'N', u'Y'] x.delete_label(u'Y') assert x.get_labels() == [u'N'] x.clear() assert len(x.get_labels()) == 0 x.load(model) y = x.classify([Datum({'x': 1}), Datum({'x': -1})]) assert [ list(sorted(z, key=lambda x: x.score, reverse=True))[0].label for z in y ] == ['Y', 'N'] assert sorted(x.get_labels()) == [u'N', u'Y'] assert json.loads(x.get_config())
def test_classifier_num(): x = Classifier(CLASSIFIER_CONFIG); x.train([ (u'Y', Datum({'x': 1})), (u'N', Datum({'x': -1})), ]) y = x.classify([ Datum({'x': 1}), Datum({'x': -1}) ]) assert [list(sorted(z, key=lambda x:x.score, reverse=True))[0].label for z in y] == ['Y', 'N'] assert sorted(x.get_labels()) == [u'N', u'Y'] model = x.dump() x.clear() assert len(x.get_labels()) == 0 x.set_label(u'Y') x.set_label(u'N') assert sorted(x.get_labels()) == [u'N', u'Y'] x.delete_label(u'Y') assert x.get_labels() == [u'N'] x.clear() assert len(x.get_labels()) == 0 x.load(model) y = x.classify([ Datum({'x': 1}), Datum({'x': -1}) ]) assert [list(sorted(z, key=lambda x:x.score, reverse=True))[0].label for z in y] == ['Y', 'N'] assert sorted(x.get_labels()) == [u'N', u'Y'] assert json.loads(x.get_config())