Untitled
unknown
plain_text
3 years ago
485 B
4
Indexable
class LogisticRegressionWithThreshold(LogisticRegression): def predict(self, X, threshold=None): if threshold == None: # If no threshold passed in, simply call the base class predict, effectively threshold=0.5 return LogisticRegression.predict(self, X) else: y_scores = LogisticRegression.predict_proba(self, X)[:, 1] y_pred_with_threshold = (y_scores >= threshold).astype(int) return y_pred_with_threshold
Editor is loading...