scikit-learn .predict() default threshold

by

The threshold can be set using clf.predict_proba()

for example:

from sklearn.tree import DecisionTreeClassifier
clf = DecisionTreeClassifier(random_state = 2)
clf.fit(X_train,y_train)
# y_pred = clf.predict(X_test)  # default threshold is 0.5
y_pred = (clf.predict_proba(X_test)[:,1] >= 0.3).astype(bool) # set threshold as 0.3

More Related Contents:

  1. Save classifier to disk in scikit-learn
  2. How to get most informative features for scikit-learn classifiers?
  3. Scikit-learn: How to obtain True Positive, True Negative, False Positive and False Negative
  4. Mixing categorial and continuous data in Naive Bayes classifier using scikit-learn
  5. Multiclass Classification
  6. How to extract the decision rules from scikit-learn decision-tree?
  7. scikit-learn & statsmodels – which R-squared is correct?
  8. Accuracy Score ValueError: Can’t Handle mix of binary and continuous target
  9. Is it possible to specify your own distance function using scikit-learn K-Means Clustering?
  10. Use scikit-learn to classify into multiple categories
  11. How to get precision, recall and f-measure from confusion matrix in Python [duplicate]
  12. LabelEncoder for categorical features?
  13. Custom transformer for sklearn Pipeline that alters both X and y
  14. Save MinMaxScaler model in sklearn
  15. Scikit Learn OneHotEncoder fit and transform Error: ValueError: X has different shape than during fitting
  16. memory issues when transforming np.array using to_categorical
  17. confusion matrix error “Classification metrics can’t handle a mix of multilabel-indicator and multiclass targets”
  18. Can anyone explain me StandardScaler?
  19. What are the pros and cons between get_dummies (Pandas) and OneHotEncoder (Scikit-learn)?
  20. How to find the importance of the features for a logistic regression model?
  21. Scikit Learn GridSearchCV without cross validation (unsupervised learning)
  22. How to compute precision, recall, accuracy and f1-score for the multiclass case with scikit learn?
  23. scikit-learn: how to scale back the ‘y’ predicted result
  24. Save Naive Bayes Trained Classifier in NLTK
  25. Tensorflow Precision / Recall / F1 score and Confusion matrix
  26. Using Smote with Gridsearchcv in Scikit-learn
  27. Does GridSearchCV perform cross-validation?
  28. Model help using Scikit-learn when using GridSearch
  29. Does the SVM in sklearn support incremental (online) learning?
  30. Determining the most contributing features for SVM classifier in sklearn

Leave a Comment