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 vote348 views1 answer
How to check the django version on a mac
Python
0 vote362 views1 answer
Python output text
Python
7 votes323 views2 answers
What is the django orm
Python
0 vote391 views1 answer
Lose your django secret key
Python
3 votes375 views2 answers
Python append row to 2d array
Python
3 votes368 views2 answers
Python sum only numbers
Python
1 vote379 views1 answer
How to save a dictionary to excel in python
Python
0 vote345 views1 answer
Correlation for specific columns
Python
11 votes376 views3 answers
Username Python
Python
10 votes436 views2 answers
Adding static file and its usage in Django
Python Django