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

5 votes149 views1 answer
Python read file without newline
Python
0 vote142 views1 answer
Plt.plot(x, softmax(scores).T, linewidth=2)
Python
1 vote145 views1 answer
How to select parts of a numpy array
Python
0 vote127 views1 answer
Webdriver.ChromeOptions()
Python
0 vote144 views1 answer
Python argparse type date
Python
0 vote184 views1 answer
Django.db.utils.ProgrammingError: relation "users" does not exist in django 3.0
Python Django
0 vote121 views1 answer
Tensorflow use growing memory
Python
0 vote221 views1 answer
Python requests login session
Python
9 votes145 views6 answers
Reaction role discord.py
Python
-1 vote166 views2 answers
Django connection cursor
Python Django