Guest User

Untitled

a guest
May 24th, 2018
95
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.81 KB | None | 0 0
  1. from sklearn.svm import LinearSVC
  2. from sklearn.calibration import CalibratedClassifierCV
  3. from sklearn import datasets
  4.  
  5. #Load iris dataset
  6. iris = datasets.load_iris()
  7. X = iris.data[:, :2] # Using only two features
  8. y = iris.target #3 classes: 0, 1, 2
  9.  
  10. linear_svc = LinearSVC() #The base estimator
  11.  
  12. # This is the calibrated classifier which can give probabilistic classifier
  13. calibrated_svc = CalibratedClassifierCV(linear_svc,
  14. method='sigmoid', #sigmoid will use Platt's scaling. Refer to documentation for other methods.
  15. cv=3)
  16. calibrated_svc.fit(X, y)
  17.  
  18.  
  19. # predict
  20. prediction_data = [[2.3, 5],
  21. [4, 7]]
  22. predicted_probs = calibrated_svc.predict_proba(prediction_data) #important to use predict_proba
  23. print predicted_probs
Add Comment
Please, Sign In to add comment