Exemplo n.º 1
0
def test_part3_1():
    """ Test from part2 should be working """
    X,Y = get_exam_dataset()
    X,Y = normalize_data(X,Y)
    log_reg = MulticlassLogisitcRegression(classes = 2)
    log_reg.fit(X,Y)
    accuracy =  evaluate(log_reg, X, Y)
    print "Accuracy ",accuracy
    assert(accuracy > 0.8)
Exemplo n.º 2
0
def test_part3_2():
    X, Y = get_MNIST_training_normalized()
    print "Normalized MNIST dataset loaded"
    ml_log_reg = MulticlassLogisitcRegression(10)
    ml_log_reg.fit(X,Y)
    print "Fitted logistic regression"
    X_test, Y_test = get_MNIST_testing_normalized()
    print "Normalized MNIST testing dataset loaded"
    accuracy = evaluate(ml_log_reg, X_test, Y_test)
    print "Accuracy on test dataset = ", accuracy
    assert(accuracy > 0.8)
    return ml_log_reg