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

4 votes367 views6 answers
Python or in if statement
Python
0 vote354 views1 answer
Python -c crypt command in python3.3 and above
Python
17 votes330 views5 answers
Python input integer
Python
0 vote345 views1 answer
Readlines replace \n
Python
0 vote307 views1 answer
Applymap(lambda x: x.encode('unicode_escape'). decode('utf-8') if isinstance(x, str) else x)
Python
0 vote344 views1 answer
Dtype cannot be bool python
Python
0 vote380 views1 answer
Scrape website with login python selenium
Python
2 votes351 views2 answers
How to install pywin32 python
Python
0 vote390 views1 answer
Python generator for reading and writing file
Python
1 vote381 views1 answer
Django model query add annotation field to show duplicate count
Python Django