Exemple #1
0
def test_1():
    np.random.seed(0)
    X = np.random.randn(200, 5)
    query_id = np.ones(len(X))
    w = np.random.randn(5)
    y = np.dot(X, w)
    coef, _ = sgd_train(X, y, query_id, 1.0, max_iter=100)
    prediction = sgd_predict(X, coef)
    tau, _ = stats.kendalltau(y, prediction)
    assert np.abs(1 - tau) > 1e-3
Exemple #2
0
def demo():
    # Basic demo test
    X_train, y_train = load_svmlight_file("demo.train", query_id=False)
    X_test, y_test = load_svmlight_file("demo.train", query_id=False)
    coef, _ = sgd_train(
        X_train, y_train, np.ones(y_test.shape), alpha=0.1, n_features=150000, model="rank", max_iter=100000
    )
    preds = sgd_predict(X_test, coef, blocks=None)
    preds = np.sign(preds)
    assert accuracy_score(y_test, preds) > 0.98
    assert precision_score(y_test, preds) > 0.98
    assert recall_score(y_test, preds) > 0.98
    assert auc_score(y_test, preds) > 0.98