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

2 votes318 views1 answer
Selenium python enter text
Python
0 vote299 views1 answer
Pie chart add outline python
Python
0 vote270 views1 answer
How to extract keys from dictreader python
Python
0 vote390 views1 answer
Django create ap
Python
79 votes300 views7 answers
Pip
Python
0 vote364 views1 answer
How to get data from multiple tables in django
Python
1 vote370 views1 answer
Factorial sequence code in python with while loops
Python
0 vote359 views1 answer
Csrf token fetch django
Python
3 votes335 views2 answers
AttributeError: module 'tensorflow' has no attribute 'placeholder'
Python
1 vote335 views2 answers
N factorial
Python