Python Classifier 예제들

예제 #1

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']

예제 #2

파일: test_classifier.py 프로젝트: yukimori/embedded-jubatus-python

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']

예제 #3

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())

예제 #4

파일: test_classifier.py 프로젝트: yukimori/embedded-jubatus-python

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())