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 vote133 views1 answer
To the power python markdown
Python
32 votes143 views3 answers
Python create new folder in path that don't exist
Python
0 vote147 views1 answer
Keras.datasets no module
Python
6 votes136 views3 answers
Django filter install
Python
0 vote132 views1 answer
Create array with shape 0,2
Python
0 vote127 views1 answer
Boto3 upload to digital digitalocean folder
Python
0 vote138 views2 answers
Spyder 32 bit download
Python
-1 vote137 views1 answer
Django custom primary key field
Python
0 vote131 views1 answer
Opencv export image
Python
0 vote133 views1 answer
First and last name generator python
Python