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 vote178 views1 answer
Flask new response style with `make_response`
Python
0 vote180 views1 answer
Open file find and replace commas python
Python
0 vote156 views1 answer
Gcsfs pd.read_parquet
Python
0 vote157 views2 answers
Logger
Python
0 vote183 views1 answer
Install os conda
Python
0 vote165 views1 answer
Reportlab python add font style
Python
1 vote178 views1 answer
Truncate date to midnight in pandas column
Python
4 votes167 views1 answer
Pandas fill empty
Python
2 votes178 views1 answer
Python dict exclude keys
Python
8 votes224 views5 answers
Python print statements
Python