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

5 votes175 views3 answers
How to print hello in python
Python
0 vote165 views1 answer
Ordenar lista decrescente python
Python
0 vote172 views1 answer
Comma separator f string python
Python
0 vote220 views1 answer
Django cms api
Python Django
0 vote220 views1 answer
Fast way to load mongodb data into python list
Python
1 vote257 views2 answers
Pyqt5 message box
Python
0 vote253 views1 answer
Pdfs in django
Python Django
7 votes187 views2 answers
Ban all discord python script
Python
0 vote145 views1 answer
Python fft frequency spectrum
Python
0 vote184 views1 answer
Strip unicode characters from strings python
Python