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 vote166 views1 answer
Limits the result count to the number specified
Python
0 vote200 views1 answer
Qspinbox value changed
Python
-2 votes177 views1 answer
What does waka waka mean
Python
0 vote170 views1 answer
Sending Emails
Python
0 vote185 views1 answer
Static Assets in Django
Python Django
5 votes175 views2 answers
Django admin password reset
Python
0 vote165 views1 answer
Python nearly equal
Python
0 vote192 views1 answer
How remove name of index pandas
Python
0 vote193 views1 answer
Sys.path[0]
Python
0 vote197 views4 answers
Insert blank row in data frame
Python