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

1 vote206 views1 answer
How to draw shape square in python turtle
Python
0 vote227 views1 answer
Python replace \n with something else
Python
0 vote208 views1 answer
Value isn't remembered in lambda python
Python
0 vote206 views2 answers
Lambda function dataframe
Python
-1 vote189 views1 answer
Matplotlib limit number of ticks
Python
0 vote233 views1 answer
Taggablemanager serializer django
Python Django
9 votes259 views2 answers
How to loop through lines python
Python
0 vote207 views1 answer
Python discord know message from bot
Python
1 vote192 views1 answer
Increase limit of recusrion python
Python
5 votes206 views2 answers
Django connexion session time
Python