From 72a3f255ca5239ff915da0693b33d1dc0e8f93cf Mon Sep 17 00:00:00 2001 From: Nikolay Laptev Date: Wed, 18 Nov 2015 00:37:53 +0000 Subject: [PATCH] Added a ranking example --- examples/ranking_example.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 examples/ranking_example.py diff --git a/examples/ranking_example.py b/examples/ranking_example.py new file mode 100644 index 00000000..e11e7d7e --- /dev/null +++ b/examples/ranking_example.py @@ -0,0 +1,30 @@ +""" +================ +Ranking example +================ +""" + +import numpy as np + +from sklearn.datasets import load_diabetes +from sklearn.utils.testing import assert_almost_equal + +from lightning.ranking import PRank +from lightning.ranking import KernelPRank + +# TODO: Add more datasets. +bunch = load_diabetes() +X, y = bunch.data, bunch.target +y = np.round(y, decimals=-2) + +# TODO: Add more rankers. +rankers = (PRank(n_iter=10, shuffle=False, random_state=0), + KernelPRank(kernel="linear", + n_iter=10, + shuffle=True, + random_state=0)) + +for ranker in rankers: + print ranker.__class__.__name__ + ranker.fit(X, y) + print ranker.score(X, y)