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 vote283 views1 answer
Docker python heelo world doesnt print anything
Python
0 vote263 views1 answer
Add colorbar to 2d hist
Python
1 vote248 views1 answer
View whole dataset in python
Python
0 vote259 views1 answer
Im save to a bytes io python
Python
0 vote253 views1 answer
Check strict superset hackerrank solution
Python
0 vote265 views1 answer
How to check if a list is nested or not
Python
2 votes249 views1 answer
String module in python
Python
0 vote263 views1 answer
Rolling call on one column and groupby second pandas
Python
0 vote265 views1 answer
Python if index not out of range
Python
0 vote256 views1 answer
F string add 0 before python
Python