Skip to content

Instantly share code, notes, and snippets.

@nkt1546789
Created July 18, 2015 03:48
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save nkt1546789/294dda4bf5e6b1b390bf to your computer and use it in GitHub Desktop.
Save nkt1546789/294dda4bf5e6b1b390bf to your computer and use it in GitHub Desktop.
demo script for scikit-multilearn. should be executed in /path/to/scikit-multilearn.
from skmultilearn.dataset import Dataset
from skmultilearn.meta.br import BinaryRelevance
from sklearn.svm import SVC
from sklearn.metrics import hamming_loss
train_set=Dataset.load_dataset_dump("skmultilearn/data/scene-train.dump.bz2")
test_set=Dataset.load_dataset_dump("skmultilearn/data/scene-test.dump.bz2")
clf=BinaryRelevance(SVC(kernel="linear"))
clf.fit(train_set["X"],train_set["y"])
ypred=clf.predict(test_set["X"])
# evaluate the performance using hamming loss. lower is better.
print hamming_loss(test_set["y"],ypred)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment