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

15 votes170 views1 answer
Importing file from another folder python
Python
0 vote180 views1 answer
ValueError: query data dimension must match training data dimension
Python
9 votes158 views6 answers
How to write a function in python
Python
1 vote162 views1 answer
Convert array to set python
Python
0 vote166 views1 answer
Python look for image on screen
Python
0 vote187 views1 answer
Def dict(d)
Python
81 votes170 views6 answers
Check python version
Python
10 votes228 views6 answers
Remove unnamed columns pandas
Python
1 vote157 views1 answer
Button position python
Python
0 vote160 views1 answer
List Get a Element-2
Python