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 vote300 views1 answer
Replace nat with date pandas
Python
0 vote292 views1 answer
List of imported modules in python
Python
0 vote317 views1 answer
Python Hewwo wowwd
Python
0 vote317 views3 answers
Django add user to group
Python Django
0 vote303 views1 answer
Flask return 404
Python
1 vote305 views1 answer
Pandas select data conditional
Python
0 vote295 views1 answer
Plot by hour of day pandas
Python
84 votes282 views6 answers
Python datetime from string
Python
0 vote300 views1 answer
Concatenating datfra,esin pandas
Python
-2 votes298 views4 answers
Tf version
Python