Advertisement
Guest User

Untitled

a guest
Mar 31st, 2020
108
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.28 KB | None | 0 0
  1. def f1_smart(y_true, y_pred):
  2. args = np.argsort(y_pred)
  3. tp = y_true.sum()
  4. fs = (tp - np.cumsum(y_true[args[:-1]])) / np.arange(y_true.shape[0] + tp - 1, tp, -1)
  5. res_idx = np.argmax(fs)
  6. return 2 * fs[res_idx], (y_pred[args[res_idx]] + y_pred[args[res_idx + 1]]) / 2
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement