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.3.tar.gz
(1.9 kB
view hashes)
Built Distribution
Close
Hashes for threshold_finder-0.1.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b02f06d6df9d1217a93b469e6d2699804cc51b3f03f4b28f4687d03e4bc527c5 |
|
MD5 | 1523d62b32a803bc7a84be1a5306c827 |
|
BLAKE2b-256 | 515b02a352567d02a9de960d89b188be0bef47b144fa8ed5f90461d5071ee305 |