Sklearn predict threshold

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

Source: stackoverflow.com

Similar codes

0 vote142 views1 answer
Formate a phonenumber in phonenumber package with PhoneNumberFormat
Python
2 votes130 views2 answers
Django-admin command not found
Python
0 vote130 views1 answer
Dictionary lookup python
Python
28 votes153 views2 answers
Add day in date python
Python
1 vote117 views1 answer
_set in django
Python
0 vote158 views1 answer
Migrations.rename_field django
Python Django
1 vote130 views2 answers
Python try except raise error
Python
0 vote137 views1 answer
Python running mean pandas
Python
4 votes154 views1 answer
Import sklearn.metrics from plot_confusion_matrix
Python
0 vote158 views1 answer
Get span text selenium python
Python