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

111 votes181 views6 answers
Python while loop
Python
0 vote304 views1 answer
Switch columns and rows python
Python
0 vote209 views1 answer
Day 14 scope hackerrank solution in python
Python
1 vote168 views1 answer
What is serialization in django
Python
0 vote175 views2 answers
Aminul
Python
0 vote207 views1 answer
Xlrd documentation
Python
3 votes183 views1 answer
Rotate picture in opencv2 python
Python
1 vote174 views1 answer
Python to mac executable
Python
1 vote181 views1 answer
How to get prime numbers in a list in python using list comprehension
Python
0 vote148 views1 answer
Locate certificate path python
Python