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 vote211 views1 answer
Change every value in a np array
Python
10 votes223 views3 answers
Add picture to jupyter notebook
Python
1 vote255 views1 answer
Solution TypeError: can’t multiply sequence by non-int of type ‘float’
Python
0 vote237 views1 answer
How to use annoy python
Python
0 vote226 views1 answer
Module level import not at top of file
Python
1 vote225 views1 answer
How to remove python3 on mac
Python
0 vote233 views1 answer
Ms access python dataframe
Python
0 vote223 views1 answer
Pandas filter on range of values
Python
0 vote226 views1 answer
Pandas : stratification (mean)
Python
46 votes237 views6 answers
Comment in python
Python