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 vote235 views1 answer
Autofield in django
Python Django
2 votes187 views1 answer
How to downgrade python to 3.7 4 anaconda
Python
16 votes203 views4 answers
List to dataframe columns
Python
0 vote190 views1 answer
Networkx - calculate degree per each node
Python
8 votes198 views1 answer
Creating dynamic variable in python
Python
0 vote217 views1 answer
Python string after character
Python
0 vote186 views1 answer
Python computer guesses your number
Python
0 vote191 views1 answer
How to download chatterbot
Python
0 vote205 views1 answer
Django queryset group by sum
Python
0 vote201 views1 answer
Spacegoat meaning
Python