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 vote323 views1 answer
Pyhton how to chnge colour of graphs
Python
0 vote364 views1 answer
Select values for row matching condition
Python
0 vote366 views1 answer
Cls in python
Python
0 vote342 views1 answer
Log scale seaborn
Python
1 vote331 views1 answer
How to make a breakable loop in python
Python
0 vote331 views2 answers
Python top label plot
Python
0 vote329 views1 answer
2 - 20 python
Python
0 vote360 views1 answer
Combining sparse class
Python
0 vote338 views1 answer
ModuleNotFoundError: No module named 'myapp' alembic
Python
0 vote340 views1 answer
Counting valleys hackerrank solution python
Python