Finding optimal threshold based on ROC CURVE.
Project description
Installation
pip install threshold-finder
Usage
Folowing display an usage example.
>>> from threshold_finder.finder import OptimalThresholdFinder, ThresholdFinder, YoudenThresholdFinder
>>> # Example data
>>> true_label = pd.Series([1,1,1,0,0,0])
>>> predicted_proba = pd.Series([0.9, 0.8, 0.7, 0.72, 0.6, 0.5])
>>> # Use a specific finder directly ...
>>> finder = YoudenThresholdFinder()
>>> optimal_threshold = finder.optimal_threshold(true_label, predicted_proba)
>>> print(optimal_threshold)
0.7
>>> # ... Or use the factory
>>> factory = ThresholdFinder()
>>> finder = factory.get_finder(method="youden_statistic")
>>> optimal_threshold = finder.optimal_threshold(true_label, predicted_proba)
>>> print(optimal_threshold)
0.7
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
threshold-finder-0.1.4.tar.gz
(1.9 kB
view hashes)
Built Distribution
Close
Hashes for threshold_finder-0.1.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ef363552324e2e2cb0b0e55e30054978c344abdc8dc834bde01a7a1de962421a |
|
MD5 | 49d82da42bf1fbd8a981b750e888b4fd |
|
BLAKE2b-256 | 020eacfb1d5073e74309ea7969d532257140a0801301bfa9be3f50faab626813 |