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

1 vote325 views1 answer
Append a dataframe to an empty dataframe
Python
3 votes330 views3 answers
Python list contains substring
Python
6 votes298 views1 answer
All tkinter events
Python
0 vote326 views1 answer
Get the value of qpushbutton pyqt5 with argument
Python
26 votes345 views5 answers
Display max rows pandas
Python
2 votes433 views1 answer
How to write a numpy array to a file in python
Python
0 vote342 views1 answer
Django querset group by sum
Python Django
0 vote346 views1 answer
Fernet: Cannot decrypt strings saved in csv with pandas
Python
0 vote375 views2 answers
Sqlalchemy function for default value for column
Python
1 vote305 views1 answer
How to make an simple python client
Python