mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-09 14:57:59 -05:00
24 lines
575 B
Python
24 lines
575 B
Python
from enum import Enum
|
|
|
|
|
|
class ImportanceStrategy(str, Enum):
|
|
"""
|
|
Strategy to use for calculating feature importances, which are used to estimate the predictive power of each feature
|
|
in training loops and explanations.
|
|
"""
|
|
|
|
SHAP = 'shap'
|
|
"""
|
|
Use SHAP (SHapley Additive exPlanations) to calculate feature importances.
|
|
"""
|
|
|
|
PERMUTATION = 'permutation'
|
|
"""
|
|
Use the permutation-based feature importances.
|
|
"""
|
|
|
|
IMPURITY = 'impurity'
|
|
"""
|
|
Use the impurity-based feature importances from the RandomForestClassifier.
|
|
"""
|