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 vote152 views1 answer
Python ip address is subnet of
Python
0 vote174 views1 answer
Matplotlib boxplot remove outliers
Python
2 votes299 views1 answer
How to write a numpy array to a file in python
Python
0 vote172 views1 answer
S = 1 + 2 + ... + n in python
Python
0 vote155 views1 answer
MASS algorithm python
Python
0 vote143 views1 answer
Check tensor type tensorflow
Python
1 vote153 views2 answers
Creating lambda expressions in comprehension list
Python
0 vote161 views1 answer
How to get one record in django
Python
5 votes144 views2 answers
Python how to put int into list
Python
27 votes149 views2 answers
Python date get day
Python