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 vote273 views1 answer
Sorting algorithms in python
Python
0 vote298 views1 answer
Tk inter entry
Python
0 vote269 views2 answers
Cannot create group in read-only mode. keras
Python
2 votes299 views1 answer
Django load model by name
Python Django
0 vote271 views1 answer
Python opens windows store
Python
5 votes269 views1 answer
Python put quotes in string
Python
17 votes289 views6 answers
Euclidean distance python
Python
0 vote312 views1 answer
Source code for making Telegram robot with Python
Python
0 vote272 views1 answer
Python write to error stream
Python
0 vote271 views1 answer
Python print functoin
Python