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

17 votes123 views4 answers
How to check python version linux
Python
12 votes116 views6 answers
Convert decimal to binary in python
Python
0 vote124 views1 answer
Mechanize python #2
Python
0 vote113 views1 answer
String + string in python
Python
0 vote133 views1 answer
App = Flask(_name_) NameError: name '_name_' is not defined
Python Flask
13 votes113 views2 answers
How to convert python input to int
Python
0 vote121 views1 answer
Python detect warning
Python
16 votes136 views3 answers
Python clear file before writing
Python
1 vote126 views1 answer
Beautiful soup 4
Python
0 vote106 views1 answer
Détruire une variable python
Python