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

0 vote120 views1 answer
Multiple arguments with multiprocessing python
Python
0 vote113 views1 answer
Roganrola
Python
0 vote151 views1 answer
What is NoReverseMatch in django?
Python
0 vote193 views2 answers
Fibbonachi python
Python
0 vote117 views1 answer
Dataflair python
Python
0 vote117 views1 answer
Assert_series_equal
Python
2 votes187 views1 answer
Django try catch exception
Python Django
0 vote116 views1 answer
Pandas groupby without reset index
Python
19 votes113 views1 answer
Find root directory of jupyter notebook
Python
0 vote135 views1 answer
Python get response from url
Python