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

1 vote294 views1 answer
Series to numpy array
Python
9 votes348 views5 answers
Get dummies pandas
Python
0 vote287 views1 answer
Python best docstring format
Python
-1 vote255 views1 answer
Django any() function raw sql query
Python
0 vote283 views2 answers
Get data from dynamodb using boto3
Python
0 vote286 views1 answer
Views.MainView.as_view(), name='all'
Python
6 votes272 views2 answers
Pygame size of image
Python
0 vote287 views2 answers
Unpacking tuples in python
Python
11 votes268 views6 answers
Np.random.randint
Python
0 vote270 views1 answer
Dataframe memory usage
Python