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 vote266 views1 answer
Python close database connection
Python
-1 vote252 views1 answer
How to separate numeric and categorical variables in python
Python
0 vote283 views1 answer
Signup class
Python
14 votes338 views6 answers
Install django
Python Django
0 vote268 views1 answer
Quando è stata inventata la lavastoviglie
Python
0 vote298 views2 answers
Python C-like structs
Python
1 vote262 views1 answer
While not command in python
Python
0 vote264 views1 answer
Load dataset X = pd.DataFrame(data.data, columns=data.features)
Python
3 votes324 views1 answer
Django import model from another app
Python Django
2 votes255 views3 answers
Convert series to datetime
Python