Advertisement
Guest User

Untitled

a guest
Aug 4th, 2015
152
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.52 KB | None | 0 0
  1. from skmultilearn.dataset import Dataset
  2. from skmultilearn.meta.br import BinaryRelevance
  3. from sklearn.svm import SVC
  4. from sklearn.metrics import hamming_loss
  5.  
  6. train_set=Dataset.load_dataset_dump("skmultilearn/data/scene-train.dump.bz2")
  7. test_set=Dataset.load_dataset_dump("skmultilearn/data/scene-test.dump.bz2")
  8.  
  9. clf=BinaryRelevance(SVC(kernel="linear"))
  10. clf.fit(train_set["X"],train_set["y"])
  11.  
  12. ypred=clf.predict(test_set["X"])
  13.  
  14. # evaluate the performance using hamming loss. lower is better.
  15. print hamming_loss(test_set["y"],ypred)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement