|
@@ -10,7 +10,7 @@ import sys
|
|
|
|
|
|
import numpy as np
|
|
|
|
|
|
-from numpy.typing import ArrayLike
|
|
|
+# from numpy.typing import ArrayLike
|
|
|
|
|
|
if (sys.version_info.major == 3) & (sys.version_info.minor >= 8):
|
|
|
from typing import Iterable, Callable
|
|
@@ -19,8 +19,10 @@ else:
|
|
|
|
|
|
|
|
|
def get_optimal_proba_threshold(score_func: Callable,
|
|
|
- y_true: ArrayLike,
|
|
|
- proba: ArrayLike,
|
|
|
+ # y_true: ArrayLike,
|
|
|
+ # proba: ArrayLike,
|
|
|
+ y_true,
|
|
|
+ proba,
|
|
|
threshold_set: Iterable = None):
|
|
|
"""
|
|
|
"""
|