Beispiel #1
0
def test_repr():

    model = VW()
    expected = "VW('convert_labels:False', 'quiet:True', 'sgd:False')"
    assert expected == model.__repr__()

    model = VWClassifier()
    expected = "VWClassifier('convert_labels:False', "\
    "'loss_function:logistic', 'quiet:True', 'sgd:False')"
    assert expected == model.__repr__()

    model = VWRegressor()
    expected = "VWRegressor('convert_labels:False', 'quiet:True', 'sgd:False')"
    assert expected == model.__repr__()

    model = VW(convert_to_vw=False,
               oaa=3,
               loss_function='logistic',
               probabilities=True)
    expected = "VW('convert_labels:False', 'loss_function:logistic', "\
    "'oaa:3', 'probabilities:True', 'quiet:True', 'sgd:False')"
    assert expected == model.__repr__()