Skip to content

Commit

Permalink
wip in brier score 2
Browse files Browse the repository at this point in the history
  • Loading branch information
Raul Casaña Eslava committed Jul 5, 2022
1 parent fc0fd94 commit 2fb1b37
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion xgbse/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def approx_brier_score(y_true, survival, aggregate="mean"):
"""
events, times = convert_y(y_true)
events = events.astype(bool)

# TODO Check this function with our data, why all are ones?
# calculating censoring distribution
censoring_dist, _, _ = calculate_kaplan_vectorized(
times.reshape(1, -1), ~events.reshape(1, -1), survival.columns
Expand Down

0 comments on commit 2fb1b37

Please sign in to comment.